Update app.py

app.py

@@ -432,45 +432,61 @@ def get_chat_list():
 
 @app.route("/api/chat", methods=["POST"])
 def chat():
-    if not llm_chain:
-        return jsonify({
-            "success": False,
-            "response": "System is still initializing. Please try again in a moment."
-        })
-    data = request.json
-    user_input = data.get("message", "")
-    session_id = data.get("sessionId", "default")
+    try:
+        # Check if LLM chain is initialized
+        if not llm_chain:
+            print("LLM chain not initialized")
+            return jsonify({
+                "success": False,
+                "response": "System is still initializing. Please try again in a moment."
+            })
+
+        # Get request data
+        data = request.json
+        if not data:
+            print("No data received in request")
+            return jsonify({
+                "success": False,
+                "response": "No message data received"
+            })
 
-    # Get or create session
-    session = ChatSession(session_id)
+        user_input = data.get("message", "")
+        session_id = data.get("sessionId", "default")
 
-    try:
-        # Add user message
+        print(f"Processing message for session {session_id}: {user_input[:100]}...")
+
+        # Get or create session
+        session = ChatSession(session_id)
+
+        # Add user message and update metadata
         session.add_message("user", user_input)
         update_chat_metadata(session_id, user_input)
 
         # Get memory variables
         memory_vars = session.get_memory_variables()
+        print("Memory variables loaded successfully")
 
         # Generate response
+        print("Generating response...")
         raw_response = llm_chain.run(
             user_request=user_input,
             chat_history=memory_vars.get("chat_history", ""),
             important_info="\n".join(session.important_info)
         )
+        print(f"Raw response generated: {raw_response[:100]}...")
 
         # Extract and store important information
         new_important_info = extract_important_info(raw_response)
         for info in new_important_info:
             session.add_important_info(info)
+            print(f"Added important info: {info}")
 
-        # Format the response properly with code block structure
+        # Format the response with code block structure
         def format_response(response):
             # First, handle code blocks with language specification
             formatted = re.sub(
                 r'```(\w+)\n(.*?)\n```',
-                lambda
-                    m: f'<div class="code-block-wrapper">\n<button class="test-button">Test Code</button>\n<button class="copy-button">Copy Code</button>\n<pre><code class="hljs {m.group(1)}">{m.group(2)}</code></pre>\n<div class="test-results"></div>\n</div>',
+                lambda m: f'<div class="code-block-wrapper">\n<button class="test-button">Test Code</button>\n<button class="copy-button">Copy Code</button>\n<pre><code class="hljs {m.group(1)}">{m.group(2)}</code></pre>\n<div class="test-results"></div>\n</div>',
                 response,
                 flags=re.DOTALL
             )
@@ -478,8 +494,7 @@ def chat():
             # Then handle code blocks without language specification
             formatted = re.sub(
                 r'```\n(.*?)\n```',
-                lambda
-                    m: f'<div class="code-block-wrapper">\n<button class="test-button">Test Code</button>\n<button class="copy-button">Copy Code</button>\n<pre><code class="hljs">{m.group(1)}</code></pre>\n<div class="test-results"></div>\n</div>',
+                lambda m: f'<div class="code-block-wrapper">\n<button class="test-button">Test Code</button>\n<button class="copy-button">Copy Code</button>\n<pre><code class="hljs">{m.group(1)}</code></pre>\n<div class="test-results"></div>\n</div>',
                 formatted,
                 flags=re.DOTALL
             )
@@ -493,22 +508,28 @@ def chat():
 
             return formatted
 
-        # Format the response
+        # Format and store the response
+        print("Formatting response...")
         formatted_response = format_response(raw_response)
-
-        # Store the formatted response
         session.add_message("assistant", formatted_response)
+        print("Response formatted and stored successfully")
 
         return jsonify({
-            "response": formatted_response,
             "success": True,
+            "response": formatted_response,
             "important_info": session.important_info
         })
 
     except Exception as e:
+        import traceback
+        error_trace = traceback.format_exc()
+        print(f"Error in chat endpoint: {str(e)}")
+        print(f"Traceback: {error_trace}")
+        
         return jsonify({
-            "response": f"An error occurred: {str(e)}",
-            "success": False
+            "success": False,
+            "response": f"An error occurred while processing your message: {str(e)}",
+            "error_details": error_trace
         })