Update app.py

app.py

@@ -432,104 +432,87 @@ def get_chat_list():
 
 @app.route("/api/chat", methods=["POST"])
 def chat():
-    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."
-            })
+    print("Entering chat endpoint")  # Log entry point
+    
+    if not llm_chain:
+        print("Error: LLM chain not initialized")
+        return jsonify({
+            "success": False,
+            "response": "System is still initializing. Please try again in a moment."
+        })
 
-        # Get request data
+    try:
+        # Log incoming request
         data = request.json
-        if not data:
-            print("No data received in request")
-            return jsonify({
-                "success": False,
-                "response": "No message data received"
-            })
-
+        print(f"Received request data: {data}")
+        
         user_input = data.get("message", "")
         session_id = data.get("sessionId", "default")
-
         print(f"Processing message for session {session_id}: {user_input[:100]}...")
 
         # Get or create session
+        print("Creating chat 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 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>',
-                response,
-                flags=re.DOTALL
-            )
-
-            # 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>',
-                formatted,
-                flags=re.DOTALL
-            )
-
-            # Handle inline code
-            formatted = re.sub(
-                r'`([^`]+)`',
-                r'<code class="inline-code">\1</code>',
-                formatted
+        try:
+            # Add user message
+            print("Adding user message...")
+            session.add_message("user", user_input)
+            update_chat_metadata(session_id, user_input)
+
+            # Get memory variables
+            print("Getting memory variables...")
+            memory_vars = session.get_memory_variables()
+            print(f"Memory variables loaded: {str(memory_vars)[:200]}...")
+
+            # Generate response
+            print("Generating LLM 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 received: {raw_response[:200]}...")
+
+            # Extract important information
+            print("Extracting important info...")
+            new_important_info = extract_important_info(raw_response)
+            print(f"Extracted important info: {new_important_info}")
+            
+            for info in new_important_info:
+                session.add_important_info(info)
+
+            # Format the response
+            print("Formatting response...")
+            formatted_response = format_response(raw_response)
+            
+            # Store the response
+            print("Storing assistant response...")
+            session.add_message("assistant", formatted_response)
+
+            print("Successfully completed chat processing")
+            return jsonify({
+                "response": formatted_response,
+                "success": True,
+                "important_info": session.important_info
+            })
 
-            return formatted
-
-        # Format and store the response
-        print("Formatting response...")
-        formatted_response = format_response(raw_response)
-        session.add_message("assistant", formatted_response)
-        print("Response formatted and stored successfully")
-
-        return jsonify({
-            "success": True,
-            "response": formatted_response,
-            "important_info": session.important_info
-        })
+        except Exception as inner_e:
+            print(f"Inner processing error: {str(inner_e)}")
+            import traceback
+            print(f"Inner error traceback: {traceback.format_exc()}")
+            raise
 
     except Exception as e:
         import traceback
         error_trace = traceback.format_exc()
         print(f"Error in chat endpoint: {str(e)}")
-        print(f"Traceback: {error_trace}")
+        print(f"Full error traceback: {error_trace}")
         
         return jsonify({
+            "response": f"An error occurred: {str(e)}",
             "success": False,
-            "response": f"An error occurred while processing your message: {str(e)}",
-            "error_details": error_trace
+            "error_trace": error_trace
         })