Add code

app.py

@@ -8,6 +8,7 @@ from typing import AsyncGenerator
 
 import asyncpg
 import gradio as gr
+from gradio.utils import get_space
 import numpy as np
 import pydantic_core
 from gradio_webrtc import (
@@ -24,6 +25,11 @@ from pydantic_ai import RunContext
 from pydantic_ai.agent import Agent
 from pydantic_ai.messages import ModelStructuredResponse, ModelTextResponse, ToolReturn
 
+if not get_space():
+    from dotenv import load_dotenv
+
+    load_dotenv()
+
 DOCS = json.load(open("gradio_docs.json"))
 
 groq_client = Groq()
@@ -38,7 +44,7 @@ class Deps:
 
 SYSTEM_PROMPT = (
     "You are an assistant designed to help users answer questions about Gradio. "
-    "You have a retrival tool that can provide relevant documentation sections based on the user query. "
+    "You have a retrieve tool that can provide relevant documentation sections based on the user query. "
     "Be curteous and helpful to the user but feel free to refuse answering questions that are not about Gradio. "
 )
 
@@ -60,7 +66,7 @@ async def database_connect(
     create_db: bool = False,
 ) -> AsyncGenerator[asyncpg.Pool, None]:
     server_dsn, database = (
-        os.getenv("DATABASE_URL"),
+        os.getenv("DB_URL"),
         "gradio_ai_rag",
     )
     if create_db:
@@ -171,7 +177,7 @@ async def stream_from_agent(
 with gr.Blocks() as demo:
     placeholder = """
 <div style="display: flex; justify-content: center; align-items: center; gap: 1rem; padding: 1rem; width: 100%">
-    <img src="/gradio_api/file=logo.svg" style="max-width: 200px; height: auto">
+    <img src="/gradio_api/file=gradio_logo.png" style="max-width: 200px; height: auto">
     <div>
         <h1 style="margin: 0 0 1rem 0">Chat with Gradio Docs 🗣️</h1>
         <h3 style="margin: 0 0 0.5rem 0">
@@ -188,7 +194,7 @@ with gr.Blocks() as demo:
         label="Gradio Docs Bot",
         type="messages",
         placeholder=placeholder,
-        avatar_images=(None, "logo.svg"),
+        avatar_images=(None, "gradio_logo.png"),
     )
     audio = WebRTC(
         label="Talk with the Agent",
@@ -210,4 +216,4 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=["logo.svg"])
+    demo.launch(allowed_paths=["gradio_logo.png"])