Update app.py

app.py

@@ -32,8 +32,8 @@ class LocalLLM:
                 out = outputs[0]  # fallback if it's just a list of strings
             else:
                 out = str(outputs)
-            # print(f"🧪 Final output string: {out[:100]}")
-            return out
+            print("🧪 Final object to return:", type(response_text), out[:100])
+            return {'role': 'assistant', 'content': out}
         except Exception as e:
             print(f"❌ Error in LocalLLM.generate(): {e}")
             raise
@@ -286,8 +286,8 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     # Test the agent
-    agent = BasicAgent()
-    agent.agent.run("What is 2+2?")
+    # agent = BasicAgent()
+    # agent.agent.run("What is 2+2?")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
     demo.launch(debug=True, share=False)