Update app.py

app.py

@@ -37,13 +37,27 @@ class MyAgent:
             {"role": "user", "content": question}
         ]
         try:
-            # Correct usage: model is callable, pass messages directly
-            response = self.model(messages)
-            return extract_final_answer(response)
-        except Exception as e:
-            import traceback
-            traceback.print_exc()
-            return f"AGENT ERROR: {e}"
+            response = self.model(messages)       
+        # If response is a dict, get the text content inside
+            if isinstance(response, dict):        
+            # Adjust depending on exact structure, commonly:
+            # {'choices': [{'message': {'content': 'the text'}}], ...}
+                choices = response.get('choices')
+                if choices and len(choices) > 0:
+                    text = choices[0].get('message', {}).get('content', '')
+                else:
+                    text = ''
+                elif isinstance(response, str):
+                text = response
+        else:
+    text = str(response)  # fallback
+        
+        return extract_final_answer(text)
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return f"AGENT ERROR: {e}"
+
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")