Update app.py

app.py

@@ -2,9 +2,12 @@ import aiohttp
 import asyncio
 import json
 import gradio as gr
+import os
 
 # Define the API URLs
-BASE_URL = "https://free-duck-duck-go-assist-coral.vercel.app/api"
+BASE_URL = os.getenv("URL")  # Ensure the environment variable "URL" is set
+if not BASE_URL:
+    raise ValueError("Environment variable 'URL' not set")
 TOKEN_URL = BASE_URL + "/get-token"
 CHAT_URL = BASE_URL + "/conversation"
 
@@ -51,7 +54,6 @@ async def chat(messList):
             messHistory.append({"role": "assistant", "content": fullmessage})  # Append assistant response
             return fullmessage
 
-
 def gradio_chat(user_input, mode):
     """Synchronous wrapper for the async chat function, integrated with Gradio."""
     messHistory.append({"role": "user", "content": f"[{mode}] {user_input}"})
@@ -60,20 +62,22 @@ def gradio_chat(user_input, mode):
     assistant_response = loop.run_until_complete(chat(messHistory))
     return assistant_response
 
-
 # Gradio interface for user interaction
+def chat_interface(user_input, mode):
+    return gradio_chat(user_input, mode)
+
 with gr.Blocks() as demo:
     gr.Markdown("# Chat with AI")
-    chat_box = gr.ChatInterface(fn=gradio_chat, label="Assistant")
-
-    # Radio button for selecting modes
-    radio_mode = gr.Radio(["Friendly", "Formal", "Humorous"], label="Chat Mode", value="Friendly")
-
-    # Chatbot UI setup with mode and user input
-    chat_ui = gr.ChatInterface(fn=lambda user_input: gradio_chat(user_input, radio_mode.value))
-
-    # Link mode selection to chat interface
-    chat_box.submit(gradio_chat, [chat_box, radio_mode], chat_box)
-
+    
+    with gr.Row():
+        radio_mode = gr.Radio(["Friendly", "Formal", "Humorous"], label="Chat Mode", value="Friendly")
+    
+    with gr.Row():
+        chatbot = gr.Interface(
+            fn=lambda user_input: chat_interface(user_input, radio_mode.value),
+            inputs=[gr.Textbox(label="Your message")],
+            outputs=[gr.Textbox(label="Assistant response")]
+        )
+    
 # Launch the Gradio app
 demo.launch()