feat: add logo on top

gradio_app.py

@@ -38,7 +38,7 @@ async def run_agent(request, history):
     image_path = ""
     load_dotenv()
     PANDAS_EXPORTS_PATH = os.getenv("PANDAS_EXPORTS_PATH")
-    generated_files = [f for f in os.listdir(PANDAS_EXPORTS_PATH) if f.beginswith("temp_chart_") and f.endswith(".png")]
+    generated_files = [f for f in os.listdir(PANDAS_EXPORTS_PATH) if f.startswith("temp_chart_") and f.endswith(".png")]
     if generated_files:
         image_path = os.path.join(PANDAS_EXPORTS_PATH, generated_files[0])
         image_markdown = image_to_base64_markdown(image_path)
@@ -59,22 +59,28 @@ async def run_agent(request, history):
 
 
 # ====================================== Gradio UI with history
-demo = gr.ChatInterface(
-    fn=run_agent,
-    type="messages",
-    flagging_mode="manual",
-    flagging_options=["Like", "Spam", "Inappropriate", "Other"],
-    examples=[
-        "List all tables in the database",
-        "Show me the schema of the dim_customer table",
-        "Count the number of active customers in 2024",
-        "Create a pie chart showing the distribution of customer statuses in the dim_customer table.",
-        "Plot a line chart showing the trend of order quantities over order dates from the dim_product_order_item table.",
-        "Generate a bar chart displaying the count of products in each product class from the dim_product table. Use beautiful and vivid colors fro visualization.",
-        "Visualize the number of product orders over time from the dim_product_order_item table, using the order_date. Show the trend monthly.",
-    ],
-    save_history=True,
-)
+LOGO_PATH = "resources/logo.png"
+with gr.Blocks() as demo:
+    gr.Image(value=LOGO_PATH, height=100, width=100, elem_id="logo")
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("<center><h1>Talk to Your Data</h1></center>")
+            gr.ChatInterface(
+                fn=run_agent,
+                type="messages",
+                flagging_mode="manual",
+                flagging_options=["Like", "Spam", "Inappropriate", "Other"],
+                examples=[
+                    "List all tables in the database",
+                    "Show me the schema of the dim_customer table",
+                    "Count the number of active customers in 2024",
+                    "Create a pie chart showing the distribution of customer statuses in the dim_customer table.",
+                    "Plot a line chart showing the trend of order quantities over order dates from the dim_product_order_item table.",
+                    "Generate a bar chart displaying the count of products in each product class from the dim_product table. Use beautiful and vivid colors fro visualization.",
+                    "Visualize the number of product orders over time from the dim_product_order_item table, using the order_date. Show the trend monthly.",
+                ],
+                save_history=True,
+            )
 
 
 # Gradio UI