Update app.py

app.py

@@ -1,10 +1,22 @@
 # app.py
 from typing import Any, Dict, List, Tuple
-from llama_index import BaseAgent  # Assurez-vous d'importer correctement votre BaseAgent
 from ansi2html import Ansi2HTMLConverter
 import gradio as gr
 from gradio.themes.utils import colors, fonts, sizes
 
+class BaseAgent:
+    """Dummy BaseAgent for demonstration purposes."""
+    
+    def stream_chat(self, message: str):
+        class ResponseGen:
+            def __iter__(self):
+                for token in message:
+                    yield token
+        return type("Response", (object,), {"response_gen": ResponseGen()})
+
+    def reset(self):
+        pass
+
 class GradioAgentChatPack:
     """Gradio chatbot to chat with your own Agent."""
 
@@ -24,6 +36,19 @@ class GradioAgentChatPack:
 
     def _generate_response(self, chat_history: List[Tuple[str, str]]) -> Tuple[str, List[Tuple[str, str]]]:
         """Generate the response from agent, and capture the stdout of the ReActAgent's thoughts."""
+        class Capturing(list):
+            def __enter__(self):
+                self._stdout = sys.stdout
+                sys.stdout = self
+                return self
+            def __exit__(self, *args):
+                sys.stdout = self._stdout
+            def write(self, x):
+                self.append(x)
+            def flush(self):
+                pass
+
+        import sys
         with Capturing() as output:
             response = self.agent.stream_chat(chat_history[-1][0])
         ansi = "\n========\n".join(output)
@@ -111,4 +136,7 @@ class GradioAgentChatPack:
 
         demo.launch(server_name="0.0.0.0", server_port=8080)
 
-# Note: Replace `BaseAgent` with your actual agent class and ensure it has the necessary methods.
+if __name__ == "__main__":
+    agent = BaseAgent()  # Instantiate your agent here
+    chat_pack = GradioAgentChatPack(agent)
+    chat_pack.run()