app.py

from flask import Flask, request, jsonify
import os

app = Flask(__name__)
UPLOAD_FOLDER = 'documents'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)

# Route for uploading documents
@app.route('/upload', methods=['POST'])
def upload_file():
    if 'file' not in request.files:
        return jsonify({"message": "No file part"}), 400
    file = request.files['file']
    if file.filename == '':
        return jsonify({"message": "No selected file"}), 400
    file.save(os.path.join(UPLOAD_FOLDER, file.filename))
    return jsonify({"message": f"File {file.filename} uploaded successfully"})

# Route for asking questions
@app.route('/ask', methods=['POST'])
def ask_question():
    data = request.get_json()
    question = data.get('question', '')
    # Placeholder logic: check your documents folder for content
    doc_path = os.path.join(UPLOAD_FOLDER, 'sample_doc.txt')
    if os.path.exists(doc_path):
        with open(doc_path, 'r') as f:
            content = f.read()
        if question.lower() in content.lower():
            answer = f"Found relevant info in sample_doc.txt: {content[:200]}..."  # return snippet
        else:
            answer = "I could not find relevant information in the document to answer your question."
    else:
        answer = "No documents uploaded."
    return jsonify({"answer": answer})

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