Spaces:
Running
Running
File size: 1,987 Bytes
2e3966c 449a596 2e3966c 449a596 2e3966c 449a596 2e3966c 449a596 2e3966c 342d4be |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import logging
import os
from flask import Flask, render_template, request, jsonify
from flask_cors import CORS
import DocumentQA
# Initialize Flask app and CORS
app = Flask(__name__)
cors = CORS(app, resources={r"/*": {"origins": "*"}})
app.logger.setLevel(logging.ERROR)
qa_system = DocumentQA.DocumentQA()
# Define the route for the index page
@app.route('/', methods=['GET'])
def index():
return render_template('index.html') # Render the index.html template
# Define the route for processing messages
@app.route('/process-message', methods=['POST'])
def process_message_route():
user_message = request.json['userMessage'] # Extract the user's message from the request
print('user_message', user_message)
bot_response = qa_system.query(user_message) # Process the user's message using the worker module
# Return the bot's response as JSON
return jsonify({
"botResponse": bot_response
}), 200
# Define the route for processing documents
@app.route('/process-document', methods=['POST'])
def process_document_route():
# Check if a file was uploaded
if 'file' not in request.files:
return jsonify({
"botResponse": "It seems like the file was not uploaded correctly, can you try "
"again. If the problem persists, try using a different file"
}), 400
file = request.files['file'] # Extract the uploaded file from the request
file_path = file.filename # Define the path where the file will be saved
file.save(file_path) # Save the file
qa_system.process_document(file_path) # Process the document using the worker module
# Return a success message as JSON
return jsonify({
"botResponse": "Thank you for providing your PDF document. I have analyzed it, so now you can ask me any "
"questions regarding it!"
}), 200
# Run the Flask app
if __name__ == "__main__":
app.run(debug=True, port=7860, host='0.0.0.0')
|