Update app.py

app.py

@@ -2,22 +2,22 @@ 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",
     "BICORP/server-3",
     "BICORP/server-4",
-    "BICORP/server-5",
-    "BICORP/server-6"
+    "BICORP/server-5"
 ]
 
 # 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,
@@ -32,7 +32,7 @@ def call_api(message, model, preset):
 css = """
 .gradio-container {
     background-color: #f0f0f0;
-    font-family: Arial, sans-serif;
+    font-family: 'Arial', sans-serif;
     padding: 20px;
     border-radius: 10px;
     box-shadow: 0 4px 10px rgba(0, 0, 0, 0.1);
@@ -89,8 +89,8 @@ def create_interface():
         
         # Settings moved to the bottom
         with gr.Accordion("Settings", open=False, elem_id="settings-container"):
-            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")
+            model = gr.Dropdown(choices=["Model A", "Model B"], label="Select Model", value="Model A")
+            preset = gr.Dropdown(choices=["Fast", "Normal", "Quality"], label="Select Preset", value="Fast")
         
         # Button click event with loading indicator
         submit_btn.click(call_api, inputs=[message, model, preset], outputs=output)