Update app.py

app.py

@@ -13,7 +13,6 @@ def format_prompt(message, history):
     return prompt
 
 def generate(prompt, history, system_prompt, theme, temperature=0.9, max_new_tokens=9048, top_p=0.95, repetition_penalty=1.0):
-
     temperature = max(float(temperature), 1e-2)
     top_p = float(top_p)
 
@@ -35,6 +34,16 @@ def generate(prompt, history, system_prompt, theme, temperature=0.9, max_new_tok
         yield output
     return output
 
+def get_theme(theme_name):
+    theme_mapping = {
+        "Base": gr.themes.Base(),
+        "Default": gr.themes.Default(),
+        "Glass": gr.themes.Glass(),
+        "Monochrome": gr.themes.Monochrome(),
+        "Soft": gr.themes.Soft()
+    }
+    return theme_mapping.get(theme_name, gr.themes.Glass())
+
 additional_inputs = [
     gr.Textbox(label="System Prompt", max_lines=1, interactive=True),
     gr.Slider(label="Temperature", value=0.9, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Higher values produce more diverse outputs"),
@@ -44,15 +53,6 @@ additional_inputs = [
     gr.Dropdown(label="Theme", choices=["Default", "Base", "Glass", "Monochrome", "Soft"], interactive=True, info="Select a theme for the app")
 ]
 
-
-theme_mapping = {
-    "Base": gr.themes.Base(),
-    "Default": gr.themes.Default(),
-    "Glass": gr.themes.Glass(),
-    "Monochrome": gr.themes.Monochrome(),
-    "Soft": gr.themes.Soft()
-}
-
 gr.ChatInterface(
     fn=generate,
     chatbot=gr.Chatbot(show_label=True, show_share_button=True, show_copy_button=True, likeable=True, layout="panel"),
@@ -60,5 +60,5 @@ gr.ChatInterface(
     title="ConvoLite",
     description="Remember! The AI might give incorrect information about people, locations, history, etc...",
     concurrency_limit=20,
-    theme=theme_mapping[additional_inputs[-1].value]  # Use the selected theme
+    theme=get_theme(additional_inputs[-1].value)  # Use the selected theme
 ).launch(show_api=False)