app.py

import random
from gradio_client import Client
import gradio as gr

# List of available servers
servers = [
    "BICORP/GOGOGOGO",
    "BICORP/server-2",
    "BICORP/server-3",
    "BICORP/server-4",
    "BICORP/server-5"
]

# 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)}"

# Custom CSS for light mode
light_css = """
.gradio-container {
    background-color: #f9f9f9;
    font-family: 'Arial', sans-serif;
    padding: 20px;
}

#chatbox {
    background-color: white;
    border-radius: 10px;
    padding: 15px;
    box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
    height: 400px;
    overflow-y: auto;
}

#input-container {
    display: flex;
    align-items: center;
    border: 1px solid #ccc;
    border-radius: 20px;
    padding: 5px;
    background-color: white;
}

#message-input {
    flex-grow: 1;
    border: none;
    padding: 10px;
    font-size: 14px;
    outline: none;
}

#send-button {
    border: none;
    background: none;
    cursor: pointer;
    padding: 5px;
    font-size: 12px;
    width: 30px;
    height: 30px;
    color: #007BFF;
    transition: opacity 0.3s;
}

#send-button:hover {
    opacity: 0.7;
}

#settings-container {
    margin-top: 20px;
}
"""

# Dark mode CSS
dark_css = """
.gradio-container {
    background-color: #1e1e1e; /* Dark background */
    color: white;              /* Light text color */
}

#chatbox {
    background-color: #2e2e2e; /* Dark chatbox */
    color: white;              /* Light text color */
}

#input-container {
    background-color: #2e2e2e; /* Dark input container */
    border: 1px solid #444;    /* Dark border */
}

#message-input {
    background-color: #3e3e3e; /* Dark input field */
    color: white;              /* Light text color */
}

#send-button {
    color: #007BFF;            /* Button text color */
}
"""

# Function to toggle dark mode
def toggle_dark_mode(is_dark):
    return dark_css if is_dark else light_css

# Create Gradio interface
def create_interface():
    with gr.Blocks() as demo:
        gr.Markdown("## 💬 Chatbot")

        # Chat display area
        chatbox = gr.Textbox(label="", interactive=False, elem_id="chatbox", lines=12)

        # Input field with button inside
        with gr.Row(elem_id="input-container"):
            message = gr.Textbox(placeholder="Type a message...", elem_id="message-input", lines=1, show_label=False)
            send_btn = gr.Button("➤", elem_id="send-button")

        # Settings section
        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 = gr.Checkbox(label="Enable Dark Mode", value=False)

        # Button click event
        send_btn.click(call_api, inputs=[message, model, preset], outputs=chatbox)

        # Change CSS based on dark mode toggle
        dark_mode.change(lambda x: demo.css.update(toggle_dark_mode(x)), inputs=[dark_mode])

    return demo

# Launch the interface
if __name__ == "__main__":
    interface = create_interface()
    interface.launch()