Spaces:
Runtime error
Runtime error
File size: 1,074 Bytes
7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 7ffcd85 8d686c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
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)
|