server.py

import logging
import os
from flask import Flask, request, jsonify
from flask_cors import CORS
import worker

app = Flask(__name__)
CORS(app, resources={r"/*": {"origins": "*"}})
app.logger.setLevel(logging.ERROR)

@app.route('/')
def index():
    return jsonify({"message": "Flask server is running!"})

@app.route('/process-message', methods=['POST'])
def process_message_route():
    user_message = request.json.get('userMessage', '')
    bot_response = worker.process_prompt(user_message)
    return jsonify({"botResponse": bot_response}), 200

@app.route('/process-document', methods=['POST'])
def process_document_route():
    if 'file' not in request.files:
        return jsonify({"botResponse": "No file uploaded."}), 400

    file = request.files['file']
    file_path = os.path.join("/tmp", file.filename)  # Use temporary directory
    file.save(file_path)

    worker.process_document(file_path)
    return jsonify({"botResponse": "PDF processed. You can now ask questions."}), 200

if __name__ == "__main__":
    app.run(debug=True, host='0.0.0.0', port=7860)