Update app.py

app.py

@@ -192,7 +192,12 @@ def stylize_reply(reply: str, persona_name: str) -> str:
     return reply
 
 
-def generate_reply(history, persona_name, tts_enabled):
+def generate_reply(history, persona_name, tts_enabled, temperature=0.8, max_tokens=120):
+    """
+    history: chatbot history with last entry [user, None].
+    persona_name: which adapter/persona to use.
+    temperature, max_tokens: UI-controlled; override persona defaults lightly.
+    """
     try:
         model.set_adapter(persona_name)
     except Exception as e:
@@ -205,12 +210,22 @@ def generate_reply(history, persona_name, tts_enabled):
     inputs = tokenizer(prompt, return_tensors="pt")
     inputs = {k: v.to(device) for k, v in inputs.items()}
 
-    params = PERSONA_GEN_PARAMS.get(persona_name, {"temperature": 0.8, "top_p": 0.9})
+    # Start from persona defaults
+    params = PERSONA_GEN_PARAMS.get(
+        persona_name, {"temperature": 0.8, "top_p": 0.9}
+    ).copy()
+
+    # Override temperature if slider is set
+    if temperature is not None:
+        params["temperature"] = float(temperature)
+
+    # Clamp / cast max_tokens
+    max_tokens = int(max_tokens) if max_tokens is not None else 120
 
     with torch.no_grad():
         output_ids = model.generate(
             **inputs,
-            max_new_tokens=120,
+            max_new_tokens=max_tokens,
             do_sample=True,
             top_p=params["top_p"],
             temperature=params["temperature"],
@@ -242,35 +257,91 @@ def generate_reply(history, persona_name, tts_enabled):
     return history, history, audio_path
 
 
-# ---------------- GRADIO UI ----------------
-with gr.Blocks() as demo:
-    gr.Markdown("## Project #3 – Multi-Character Chatbot")
+# ---------------- GRADIO UI (UPDATED) ----------------
+
+# Custom CSS for UTRGV orange theme
+custom_css = """
+.gradio-container {
+    background: #1a1a1a !important;
+}
+
+h1, h2, h3 {
+    color: #FF6600 !important;
+}
+
+label {
+    color: #FF6600 !important;
+}
+
+.message.user {
+    background: #FF6600 !important;
+}
+
+input[type="range"] {
+    accent-color: #FF6600 !important;
+}
+
+input:focus, textarea:focus, select:focus {
+    border-color: #FF6600 !important;
+}
+"""
+
+with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as demo:
+    gr.Markdown("# Multi-Personality AI Chatbot")
 
     with gr.Row():
         persona_dropdown = gr.Dropdown(
             choices=list(ADAPTER_PATHS.keys()),
-            value="Sunny Extrovert",
-            label="Choose Character",
+            value=first_persona,
+            label="Select Personality",
+        )
+        tts_checkbox = gr.Checkbox(label="Enable Text-to-Speech", value=False)
+
+    chat = gr.Chatbot(label="Conversation")
+
+    msg = gr.Textbox(
+        label="Your message",
+        placeholder="Type your message...",
+    )
+
+    with gr.Row():
+        temperature = gr.Slider(
+            minimum=0.1,
+            maximum=1.5,
+            value=0.8,
+            step=0.1,
+            label="Temperature",
+        )
+        max_tokens = gr.Slider(
+            minimum=50,
+            maximum=500,
+            value=120,
+            step=10,
+            label="Max Tokens",
         )
-        tts_checkbox = gr.Checkbox(label="Text-to-Audio", value=False)
 
-    chat = gr.Chatbot(label="Chat History")
-    msg = gr.Textbox(label="Your message")
-    audio_out = gr.Audio(label="Voice", autoplay=True)
+    audio_out = gr.Audio(label="Audio Response", autoplay=True)
     clear_btn = gr.Button("Clear Chat")
 
     def user_submit(user_message, history):
         history = history or []
+        if not user_message.strip():
+            return "", history
         return "", history + [[user_message, None]]
 
     msg.submit(
-        user_submit, [msg, chat], [msg, chat], queue=False
+        user_submit,
+        [msg, chat],
+        [msg, chat],
+        queue=False,
     ).then(
-        generate_reply, [chat, persona_dropdown, tts_checkbox], [chat, chat, audio_out]
+        generate_reply,
+        [chat, persona_dropdown, tts_checkbox, temperature, max_tokens],
+        [chat, chat, audio_out],
     )
 
     clear_btn.click(lambda: ([], None), outputs=[chat, audio_out])
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False, server_name="127.0.0.1", show_error=True, inbrowser=True)