Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 from huggingface_hub import InferenceClient
 import random
+from typing import Generator, Dict, List, Tuple, Optional
 
 # Get token from environment variable
 hf_token = os.environ.get("HF_TOKEN")
@@ -128,10 +129,10 @@ def format_history_for_gradio(history_tuples):
 # 4. Add input validation
 def respond(
     message: str, 
-    chat_history: list[tuple[str, str]], 
-    genre: str | None = None,   # Make optional type more explicit
+    chat_history: List[Tuple[str, str]], 
+    genre: Optional[str] = None,
     use_full_memory: bool = True
-) -> Generator[list[dict[str, str]], None, None]:   # More specific return type
+) -> Generator[List[Dict[str, str]], None, None]:
     """Generate a response based on the current message and conversation history."""
     if not message.strip():
         return chat_history
@@ -258,14 +259,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
             # 1) We'll return a list of 4 dicts, each dict updating 'value' & 'visible'
             def update_starter_buttons(selected_genre):
+                """Update starter buttons with examples for the selected genre."""
                 examples = get_examples_for_genre(selected_genre)
                 results = []
                 for i in range(4):
                     if i < len(examples):
-                        results.append({"value": examples[i], "visible": True})
+                        # Return just the string value instead of a dict
+                        results.append(examples[i])
                     else:
-                        results.append({"value": "(no starter)", "visible": False})
-                return tuple(results)  # we must return a tuple/list with 4 items for 4 outputs
+                        results.append("")  # Empty string for hidden buttons
+                return tuple(results)  # Return tuple of strings
 
             # 2) Initialize them with "fantasy" so they don't stay "Starter X" on page load
             # We'll just call the function and store the results in a variable, then apply them in a .load() event