Update app.py

app.py

@@ -628,11 +628,20 @@ def send_to_qwq(prompt: str):
         
         # Call the agent_run endpoint
         response = client.predict(
-            _chatbot=result[1],
+            _chatbot=result[1],  # This is correct
             api_name="/agent_run"
         )
+
+        if isinstance(response, list) and len(response) > 0:
+            # Extract text from first message in chat history
+            if isinstance(response[0], list) and len(response[0]) > 0:
+                if isinstance(response[0][1], dict):
+                    return response[0][1].get('text', 'No response text from QwQ')
+                elif isinstance(response[0][1], str):
+                    return response[0][1]
+            return 'No valid response from QwQ'
         
-        return response[0].get('text', 'No response from QwQ')
+        return 'No response from QwQ'
         
     except Exception as e:
         logging.error(f"QwQ API error: {e}")