ollama - 💡(How to fix) Fix API support for attaching files (PDF, JSON, images, bytes/base64) like the UI [1 participants]

import os
from flask import Flask, request, jsonify
from PyPDF2 import PdfReader
import json

app = Flask(__name__)

@app.route('/process', methods=['POST'])
def process_prompt():
    # Get the prompt and attached file
    prompt = request.form['prompt']
    file = request.files['file']

    # Detect file type and process accordingly
    if file.filename.endswith('.pdf'):
        pdf_reader = PdfReader(file)
        text = ''
        for page in pdf_reader.pages:
            text += page.extract_text()
        # Include the extracted text in the prompt context
        context = {'prompt': prompt, 'text': text}
    elif file.filename.endswith('.json'):
        data = json.load(file)
        # Include the JSON data in the prompt context
        context = {'prompt': prompt, 'data': data}
    else:
        # Handle other file types or errors
        return jsonify({'error': 'Unsupported file type'}), 400

    # Call the Ollama API with the updated context
    # ...

    return jsonify({'result': 'Success'}), 200

if __name__ == '__main__':
    app.run(debug=True)