Update app.py

app.py

@@ -1,83 +1,70 @@
 import random
-from gradio_client import Client
 import gradio as gr
+from gradio_client import Client
 
-# List of available servers
+# List of server endpoints
 servers = [
-    "BICORP/GOGOGOGO",
-    "BICORP/server-2",
-    "BICORP/server-3",
-    "BICORP/server-4",
-    "BICORP/server-5"
+    "BICORP/GOGOGOGO",  # Server 1
+    "BICORP/server-2",  # Server 2
+    "BICORP/server-3",  # Server 3
+    "BICORP/server-4",  # Server 4
+    "BICORP/server-5",  # Server 5
+    "BICORP/server-6"   # Server 6
 ]
 
-# Function to call the API with correct parameters
-def call_api(message, model, preset):
-    selected_server = random.choice(servers)
-    client = Client(selected_server)
-    
-    try:
-        result = client.predict(
-            message,
-            model,
-            preset,
-            api_name="/chat"
-        )
-        return result
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Dark mode CSS
-dark_css = """
-.gradio-container {
-    background-color: #1e1e1e; /* Dark background */
-    color: white;              /* Light text color */
+# Define presets for each model
+presets = {
+    "Fast": "Fast",
+    "Normal": "Normal",
+    "Quality": "Quality",
+    "Unreal Performance": "Unreal Performance"
 }
-#chatbox {
-    background-color: #2e2e2e; /* Dark chatbox */
-    color: white;              /* Light text color */
-}
-"""
-
-# Function to toggle dark mode
-def toggle_dark_mode(is_dark):
-    return dark_css if is_dark else ""
 
-# Create Gradio interface
-def create_interface():
-    with gr.Blocks(css=dark_css) as demo:
-        gr.Markdown("## 💬 Chatbot")
+# Function to respond to user input
+def respond(message, history: list, selected_model, selected_preset):
+    # Randomly select a server
+    server = random.choice(servers)
+    client = Client(server)
 
-        # Chat display area
-        chatbox = gr.Textbox(label="", interactive=False, elem_id="chatbox", lines=12)
+    # Ensure history is a list of dictionaries
+    messages = [{"role": "user", "content": message}]
 
-        # Input area
-        with gr.Row():
-            user_input = gr.Textbox(label="Type your message here...", placeholder="Enter your message...", interactive=True, elem_id="user_input")
-            send_button = gr.Button("Send")
+    # Get the response from the model
+    try:
+        response = client.predict(
+            message=message,  # Required parameter
+            param_2=selected_model,  # Model selection
+            param_3=selected_preset,  # Preset selection
+            api_name="/chat"  # Ensure this is the correct API endpoint
+        )
 
-        # Settings section
-        with gr.Accordion("⚙️ Settings", open=False):
-            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 = gr.Checkbox(label="Enable Dark Mode", value=False)
+        return response  # Return the response
+    except Exception as e:
+        return f"Error: {str(e)}"  # Return the error message
 
-        # Change CSS based on dark mode toggle
-        dark_mode.change(lambda x: demo.css.update(toggle_dark_mode(x)), inputs=[dark_mode])
+# Model names and their pseudonyms
+model_choices = [
+    ("Lake 1 Base", "Lake 1 Base")  # Only one model as per your example
+]
 
-        # Define the action for the send button
-        def send_message(message):
-            if message:
-                response = call_api(message, model.value, preset.value)
-                return f"You: {message}\nBot: {response}"
-            return ""
+# Convert pseudonyms to model names for the dropdown
+pseudonyms = [model[0] for model in model_choices]
 
-        # Connect the send button to the send_message function
-        send_button.click(send_message, inputs=user_input, outputs=chatbox)
+# Function to handle model selection and pseudonyms
+def respond_with_pseudonym(message, history: list, selected_model, selected_preset):
+    # Call the existing respond function
+    response = respond(message, history, selected_model, selected_preset)
+    
+    return response
 
-    return demo
+# Gradio Chat Interface
+demo = gr.ChatInterface(
+    fn=respond_with_pseudonym,
+    additional_inputs=[
+        gr.Dropdown(choices=pseudonyms, label="Select Model", value=pseudonyms[0]),  # Pseudonym selection dropdown
+        gr.Dropdown(choices=list(presets.keys()), label="Select Preset", value="Fast")  # Preset selection dropdown
+    ],
+)
 
-# Launch the interface
 if __name__ == "__main__":
-    interface = create_interface()
-    interface.launch()
+    demo.launch()