Create app.py

app.py

@@ -0,0 +1,114 @@
+# 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 GradioAgentChatPack:
+    """Gradio chatbot to chat with your own Agent."""
+
+    def __init__(self, agent: BaseAgent, **kwargs: Any) -> None:
+        """Init params."""
+        self.agent = agent
+        self.thoughts = ""
+        self.conv = Ansi2HTMLConverter()
+
+    def get_modules(self) -> Dict[str, Any]:
+        """Get modules."""
+        return {"agent": self.agent}
+
+    def _handle_user_message(self, user_message, history):
+        """Handle the user submitted message. Clear message box, and append to the history."""
+        return "", [*history, (user_message, "")]
+
+    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."""
+        with Capturing() as output:
+            response = self.agent.stream_chat(chat_history[-1][0])
+        ansi = "\n========\n".join(output)
+        html_output = self.conv.convert(ansi)
+        for token in response.response_gen:
+            chat_history[-1][1] += token
+            yield chat_history, str(html_output)
+
+    def _reset_chat(self) -> Tuple[str, str]:
+        """Reset the agent's chat history. And clear all dialogue boxes."""
+        self.agent.reset()
+        return "", "", ""  # clear textboxes
+
+    def run(self, *args: Any, **kwargs: Any) -> Any:
+        """Run the pipeline."""
+        llama_theme = gr.themes.Soft(
+            primary_hue=colors.purple,
+            secondary_hue=colors.pink,
+            neutral_hue=colors.gray,
+            spacing_size=sizes.spacing_md,
+            radius_size=sizes.radius_md,
+            text_size=sizes.text_lg,
+            font=(
+                fonts.GoogleFont("Quicksand"),
+                "ui-sans-serif",
+                "sans-serif",
+            ),
+            font_mono=(
+                fonts.GoogleFont("IBM Plex Mono"),
+                "ui-monospace",
+                "monospace",
+            ),
+        )
+        llama_theme.set(
+            body_background_fill="#FFFFFF",
+            body_background_fill_dark="#000000",
+            button_primary_background_fill="linear-gradient(90deg, *primary_300, *secondary_400)",
+            button_primary_background_fill_hover="linear-gradient(90deg, *primary_200, *secondary_300)",
+            button_primary_text_color="white",
+            button_primary_background_fill_dark="linear-gradient(90deg, *primary_600, *secondary_800)",
+            slider_color="*secondary_300",
+            slider_color_dark="*secondary_600",
+            block_title_text_weight="600",
+            block_border_width="3px",
+            block_shadow="*shadow_drop_lg",
+            button_shadow="*shadow_drop_lg",
+            button_large_padding="32px",
+        )
+
+        demo = gr.Blocks(
+            theme=llama_theme,
+            css="#box { height: 420px; overflow-y: scroll !important} #logo { align-self: right }",
+        )
+        with demo:
+            with gr.Row():
+                gr.Markdown(
+                    "# Gradio Chat With Your Agent Powered by LlamaIndex and LlamaHub 🦙\n"
+                    "This Gradio app allows you to chat with your own agent (`BaseAgent`).\n"
+                )
+                gr.Markdown(
+                    "[![Alt text](https://d3ddy8balm3goa.cloudfront.net/other/llama-index-light-transparent-sm-font.svg)](https://llamaindex.ai)",
+                    elem_id="logo",
+                )
+            with gr.Row():
+                chat_window = gr.Chatbot(
+                    label="Message History",
+                    scale=3,
+                )
+                console = gr.HTML(elem_id="box")
+            with gr.Row():
+                message = gr.Textbox(label="Write A Message", scale=4)
+                clear = gr.ClearButton()
+
+            message.submit(
+                self._handle_user_message,
+                [message, chat_window],
+                [message, chat_window],
+                queue=False,
+            ).then(
+                self._generate_response,
+                chat_window,
+                [chat_window, console],
+            )
+            clear.click(self._reset_chat, None, [message, chat_window, console])
+
+        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.