Update app.py

app.py

@@ -250,40 +250,47 @@ def create_app():
 
     @flask_app.route("/generate", methods=["POST"])
     def generate():
-        socketio.emit("log", {"message": "[STEP]: Entering query_gen..."})
-        data = request.json
-        prompt = data.get("prompt", "")
-        socketio.emit("log", {"message": f"[INFO]: Received prompt: {prompt}\n"})
-        # Run the agent in a separate thread
-        thread = threading.Thread(target=run_agent, args=(prompt, socketio))
-        thread.start()
-        return "OK", 200
+        try:
+            socketio.emit("log", {"message": "[STEP]: Entering query_gen..."})
+            data = request.json
+            prompt = data.get("prompt", "")
+            socketio.emit("log", {"message": f"[INFO]: Received prompt: {prompt}\n"})
+            # Run the agent in a separate thread
+            thread = threading.Thread(target=run_agent, args=(prompt, socketio))
+            thread.start()
+            return "OK", 200
+        except Exception as e:
+            socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
 
     @flask_app.route("/upload", methods=["POST", "GET"])
     def upload():
-        if request.method == 'POST':
-            file = request.files.get('file')
-            if not file:
-                print("No file uploaded")
-                return "No file uploaded", 400
-            if file and file.filename.endswith('.db'):
-                # Use flask_app.config instead of app.config
-                db_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], 'uploaded.db')
-                print("file uploaded")
-                file.save(db_path)
-                
-                # Convert the file path to an absolute path and reinitialize the agent_app
-                abs_file_path = os.path.abspath(db_path)
-                global agent_app
-                agent_app = create_agent_app(abs_file_path)
-                
-                # Use an f-string properly to log the uploaded file name
-                print(f"[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
-                socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
-                print("Received file:", file.filename)
-                return redirect(url_for("index"))  # Go back to index page
-        # For GET requests, simply render the upload form.
-        return render_template("upload.html")
+        try:
+            if request.method == 'POST':
+                file = request.files.get('file')
+                if not file:
+                    print("No file uploaded")
+                    return "No file uploaded", 400
+                if file and file.filename.endswith('.db'):
+                    # Use flask_app.config instead of app.config
+                    db_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], 'uploaded.db')
+                    print("file uploaded")
+                    file.save(db_path)
+                    
+                    # Convert the file path to an absolute path and reinitialize the agent_app
+                    abs_file_path = os.path.abspath(db_path)
+                    global agent_app
+                    agent_app = create_agent_app(abs_file_path)
+                    
+                    # Use an f-string properly to log the uploaded file name
+                    print(f"[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
+                    socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
+                    print("Received file:", file.filename)
+                    return redirect(url_for("index"))  # Go back to index page
+            # For GET requests, simply render the upload form.
+            return render_template("upload.html")
+        except Exception as e:
+            socketio.emit("log", {"message": f"[ERROR]: {str(e)}"})
+            return render_template("upload.html")
 
     return flask_app, socketio