Update app.py

app.py

@@ -9,6 +9,13 @@ app = Flask(__name__)
 cors = CORS(app, resources={r"/*": {"origins": "*"}})
 app.logger.setLevel(logging.ERROR)
 
+# Ensure worker is initialized
+worker.init_llm()
+
+# Create an uploads directory if it doesn't exist
+UPLOAD_FOLDER = "./uploads"
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
 # Define the route for the index page
 @app.route('/', methods=['GET'])
 def index():
@@ -17,15 +24,15 @@ def index():
 # 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)
-
+    user_message = request.json.get('userMessage', '')  # Extract the user's message from the request
+    
+    if not user_message:
+        return jsonify({"botResponse": "Please enter a valid message."}), 400
+    
     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
+    return jsonify({"botResponse": bot_response}), 200
 
 # Define the route for processing documents
 @app.route('/process-document', methods=['POST'])
@@ -33,13 +40,11 @@ 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"
+            "botResponse": "It seems like the file was not uploaded correctly. Please try again."
         }), 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_path = os.path.join(UPLOAD_FOLDER, file.filename)  # Save file in the uploads directory
     file.save(file_path)  # Save the file
 
     worker.process_document(file_path)  # Process the document using the worker module
@@ -52,4 +57,4 @@ def process_document_route():
 
 # Run the Flask app
 if __name__ == "__main__":
-    app.run(debug=True, port=8000, host='0.0.0.0')
+    app.run(debug=True, port=8000, host='0.0.0.0')