Update app.py

app.py

@@ -1,34 +1,55 @@
 import logging
 import os
-from flask import Flask, request, jsonify
+from flask import Flask, render_template, request, jsonify
 from flask_cors import CORS
-import worker
+import worker  # Import the worker module
 
+# Initialize Flask app and CORS
 app = Flask(__name__)
-CORS(app, resources={r"/*": {"origins": "*"}})
+cors = CORS(app, resources={r"/*": {"origins": "*"}})
 app.logger.setLevel(logging.ERROR)
 
-@app.route('/')
+# Define the route for the index page
+@app.route('/', methods=['GET'])
 def index():
-    return jsonify({"message": "Flask server is running!"})
+    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.get('userMessage', '')
-    bot_response = worker.process_prompt(user_message)
-    return jsonify({"botResponse": bot_response}), 200
+    user_message = request.json['userMessage']  # Extract the user's message from the request
+    print('user_message', user_message)
 
+    bot_response = worker.process_prompt(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": "No file uploaded."}), 400
+        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
 
-    file = request.files['file']
-    file_path = os.path.join("/tmp", file.filename)  # Use temporary directory
-    file.save(file_path)
+    worker.process_document(file_path)  # Process the document using the worker module
 
-    worker.process_document(file_path)
-    return jsonify({"botResponse": "PDF processed. You can now ask questions."}), 200
+    # 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, host='0.0.0.0', port=7860)
+    app.run(debug=True, port=8000, host='0.0.0.0')