Update app.py

app.py

@@ -2,7 +2,7 @@ import random
 from gradio_client import Client
 import gradio as gr
 
-# List of available servers
+# List of available servers (replace with actual server endpoints)
 servers = [
     "BICORP/GOGOGOGO",
     "BICORP/server-2",
@@ -13,10 +13,11 @@ servers = [
 
 # Function to call the API with error handling
 def call_api(message, model, preset):
-    selected_server = random.choice(servers)  
-    client = Client(selected_server)  
-
+    selected_server = random.choice(servers)  # Randomly select a server
+    client = Client(selected_server)  # Create a client for the selected server
+    
     try:
+        # Call the /chat endpoint with the provided parameters
         result = client.predict(
             message=message,
             param_2=model,
@@ -27,90 +28,119 @@ def call_api(message, model, preset):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Function to toggle dark mode
-def toggle_theme(dark_mode):
-    return "dark-mode" if dark_mode else "light-mode"
-
-# Custom CSS
+# Custom CSS for styling
 css = """
-.light-mode .gradio-container {
-    background-color: #FFFFFF;
-    font-family: 'Inter', sans-serif;
-    padding: 25px;
-    border-radius: 12px;
-    color: #000000;
+/* Light Mode */
+:root {
+    --bg-color: #ffffff;
+    --text-color: #333;
+    --input-bg: #f8f8f8;
+    --border-color: #ddd;
+    --button-bg: #007bff;
+    --button-text: white;
 }
 
-.dark-mode .gradio-container {
-    background-color: #121212;
-    color: #FFFFFF;
+/* Dark Mode */
+[data-theme='dark'] {
+    --bg-color: #1e1e1e;
+    --text-color: #f8f8f8;
+    --input-bg: #333;
+    --border-color: #555;
+    --button-bg: #0d6efd;
+    --button-text: white;
 }
 
-h2 {
-    text-align: center;
-    margin-bottom: 15px;
+body {
+    background-color: var(--bg-color);
+    color: var(--text-color);
 }
 
-.gr-button {
-    background-color: #1E88E5;
-    color: white;
-    font-weight: bold;
-    border-radius: 8px;
-    padding: 8px 16px;
-    transition: 0.3s;
+.gradio-container {
+    max-width: 600px;
+    margin: auto;
+    padding: 20px;
 }
 
-.gr-button:hover {
-    background-color: #1565C0;
+.chatbox {
+    border: 1px solid var(--border-color);
+    border-radius: 10px;
+    padding: 10px;
+    background-color: var(--input-bg);
+    min-height: 300px;
+    overflow-y: auto;
 }
 
-.response-box {
-    background-color: #F9F9F9;
-    border-radius: 8px;
-    padding: 15px;
-    height: 250px;
-    overflow-y: auto;
+.input-row {
+    display: flex;
+    align-items: center;
+    border: 1px solid var(--border-color);
+    border-radius: 10px;
+    padding: 5px;
+    background-color: var(--input-bg);
+}
+
+#message-input {
+    flex-grow: 1;
+    border: none;
+    background: transparent;
+    color: var(--text-color);
 }
 
-.dark-mode .response-box {
-    background-color: #1E1E1E;
-    border: 1px solid #333;
+#send-button {
+    background-color: var(--button-bg);
+    color: var(--button-text);
+    border-radius: 50%;
+    width: 40px;
+    height: 40px;
+    display: flex;
+    align-items: center;
+    justify-content: center;
 }
 
-.settings-container {
-    margin-top: 20px;
-    padding: 15px;
-    border-radius: 8px;
+#send-button:hover {
+    opacity: 0.8;
 }
 
-.dark-mode .settings-container {
-    background-color: #1E1E1E;
+#settings-container {
+    border: 1px solid var(--border-color);
+    border-radius: 10px;
+    padding: 10px;
+    margin-top: 10px;
 }
 """
 
+# Function to toggle dark mode
+def toggle_dark_mode(mode):
+    return gr.update(theme="dark" if mode else "default")
+
 # Create Gradio interface
 def create_interface():
     with gr.Blocks(css=css) as demo:
-        dark_mode = gr.State(False)  # Store dark mode state
-
-        with gr.Row():
-            gr.Markdown("## 💬 Chatbot", elem_classes=["chat-title"])
-
-        with gr.Row():
-            message = gr.Textbox(placeholder="Type a message...", elem_id="message-input", lines=1)
+        gr.Markdown("## 💬 Chatbot")
+
+        # Chatbox
+        chat_output = gr.HTML("", elem_classes="chatbox")
+
+        # Input container for message and send button
+        with gr.Row(elem_classes="input-row"):
+            message = gr.Textbox(
+                placeholder="Type a message...", 
+                show_label=False, 
+                elem_id="message-input"
+            )
             submit_btn = gr.Button("➤", elem_id="send-button")
 
-        output = gr.Textbox(interactive=False, elem_id="response-box", lines=10, show_label=False)
-
+        # Settings Panel
         with gr.Accordion("⚙️ Settings", open=False, elem_id="settings-container"):
-            model = gr.Dropdown(choices=["Lake 1 Base"], label="Model", value="Lake 1 Base")
-            preset = gr.Dropdown(choices=["Fast", "Normal", "Quality", "Unreal Performance"], label="Preset", value="Fast")
-            dark_mode_toggle = gr.Checkbox(label="Dark Mode", value=False)
+            model = gr.Dropdown(choices=["Lake 1 Base"], label="Select Model", value="Lake 1 Base")
+            preset = gr.Dropdown(choices=["Fast", "Normal", "Quality", "Unreal Performance"], label="Select Preset", value="Fast")
+            dark_mode = gr.Checkbox(label="Dark Mode", value=False)
 
-        # Apply dark mode toggle
-        dark_mode_toggle.change(toggle_theme, inputs=[dark_mode_toggle], outputs=[demo])
+        # Button click event
+        submit_btn.click(call_api, inputs=[message, model, preset], outputs=chat_output)
 
-        submit_btn.click(call_api, inputs=[message, model, preset], outputs=output)
+        # Toggle dark mode
+        dark_mode.change(toggle_dark_mode, inputs=[dark_mode])
 
     return demo