Update app.py

app.py

@@ -17,41 +17,74 @@ if not hf_token:
 client = InferenceClient("Qwen/Qwen2.5-72B-Instruct", token=hf_token)
 
 TOPIC_EXAMPLES = {
-    "Daily Life": [
-        "What do you usually do after work?",
-        "Do you like cooking? What’s your favorite dish?",
-        "Tell me about your morning routine."
-    ],
-    "Travel": [
-        "Have you ever been to another country?",
-        "What's your dream vacation?",
-        "What do you like to pack in your suitcase?"
-    ],
-    "Food": [
-        "What's your favorite food?",
-        "Can you describe how to cook a simple dish?",
-        "What did you eat today?"
-    ],
-    "Work & School": [
-        "What do you do for work or study?",
-        "Describe your typical workday.",
-        "What subjects do you like or dislike?"
-    ],
-    "Hobbies": [
-        "What do you like to do in your free time?",
-        "Tell me about a hobby you enjoy.",
-        "What’s something new you’d like to learn?"
-    ],
-    "Shopping": [
-        "What do you usually buy when you go shopping?",
-        "Do you prefer online or in-store shopping?",
-        "Tell me about your last shopping trip."
-    ],
-    "Weather": [
-        "What’s the weather like today?",
-        "Do you prefer hot or cold weather?",
-        "Tell me about your favorite season."
-    ]
+    "Daily Life": {
+        "beginner": [
+            "What time do you wake up?",
+            "Do you go to school or work?",
+            "What do you eat for breakfast?"
+        ],
+        "intermediate": [
+            "What do you usually do after work?",
+            "Do you like cooking? What’s your favorite dish?",
+            "Tell me about your morning routine."
+        ],
+        "advanced": [
+            "How do you balance personal and professional responsibilities?",
+            "What does a productive day look like for you?",
+            "How has your daily routine changed over the years?"
+        ]
+    },
+    "Travel": {
+        "beginner": [
+            "Do you like to travel?",
+            "Have you been to another city?",
+            "Do you like airplanes?"
+        ],
+        "intermediate": [
+            "Have you ever been to another country?",
+            "What's your dream vacation?",
+            "What do you like to pack in your suitcase?"
+        ],
+        "advanced": [
+            "How has travel influenced your worldview?",
+            "What do you think makes a destination culturally significant?",
+            "Describe your most challenging travel experience."
+        ]
+    },
+    "Food": {
+        "beginner": [
+            "Do you like fruit?",
+            "What’s your favorite snack?",
+            "Do you drink tea or coffee?"
+        ],
+        "intermediate": [
+            "What's your favorite food?",
+            "Can you describe how to cook a simple dish?",
+            "What did you eat today?"
+        ],
+        "advanced": [
+            "How do food choices reflect cultural identity?",
+            "What are your thoughts on sustainable eating?",
+            "Describe a memorable meal and why it stood out."
+        ]
+    },
+    "Work & School": {
+        "beginner": [
+            "Do you go to school?",
+            "Do you have a job?",
+            "Do you have homework?"
+        ],
+        "intermediate": [
+            "What do you do for work or study?",
+            "Describe your typical workday.",
+            "What subjects do you like or dislike?"
+        ],
+        "advanced": [
+            "What are the most important skills in your field?",
+            "How do you manage work-life balance?",
+            "Describe a major accomplishment and how you achieved it."
+        ]
+    }
 }
 
 MAX_HISTORY_LENGTH = 20
@@ -60,8 +93,8 @@ MAX_TOKENS = 1024
 TEMPERATURE = 0.7
 TOP_P = 0.95
 
-def get_examples_for_topic(topic):
-    return TOPIC_EXAMPLES.get(topic, TOPIC_EXAMPLES["Daily Life"])
+def get_examples_for_topic(topic, difficulty):
+    return TOPIC_EXAMPLES.get(topic, {}).get(difficulty, [])
 
 def get_conversation_prompt():
     return """You are JoJo, a friendly and supportive AI who helps people practice speaking English through fun, casual conversation.
@@ -132,7 +165,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
     gr.Markdown("""
     # 🐰 JoJo - Your Speaking Buddy
     **Practice English by having natural conversations with JoJo.** 
-    Choose a topic and start chatting. JoJo will keep the conversation going and gently correct you if needed.
+    Choose a topic and difficulty, then start chatting. JoJo will keep the conversation going and gently correct you if needed.
     """)
 
     avatar = get_avatar_url()
@@ -165,14 +198,20 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
                 value="Daily Life"
             )
 
+            difficulty = gr.Dropdown(
+                choices=["beginner", "intermediate", "advanced"],
+                label="Select Difficulty",
+                value="intermediate"
+            )
+
             gr.Markdown("## Quick Starters")
             starter_btn1 = gr.Button("Starter 1", scale=1, min_width=250, elem_classes="compact-btn")
             starter_btn2 = gr.Button("Starter 2", scale=1, min_width=250, elem_classes="compact-btn")
             starter_btn3 = gr.Button("Starter 3", scale=1, min_width=250, elem_classes="compact-btn")
             starter_buttons = [starter_btn1, starter_btn2, starter_btn3]
 
-            def update_starters(selected_topic):
-                examples = get_examples_for_topic(selected_topic)
+            def update_starters(selected_topic, selected_difficulty):
+                examples = get_examples_for_topic(selected_topic, selected_difficulty)
                 results = [examples[i] if i < len(examples) else "" for i in range(3)]
                 return tuple(results)
 
@@ -188,15 +227,16 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
             for btn in starter_buttons:
                 btn.click(fn=use_starter, inputs=[btn, chatbot, topic, memory_flag], outputs=[msg, chatbot], queue=True)
 
-            topic.change(fn=update_starters, inputs=[topic], outputs=starter_buttons)
+            topic.change(fn=update_starters, inputs=[topic, difficulty], outputs=starter_buttons)
+            difficulty.change(fn=update_starters, inputs=[topic, difficulty], outputs=starter_buttons)
 
     msg.submit(fn=respond, inputs=[msg, chatbot, topic, memory_flag], outputs=[msg, chatbot])
     submit.click(fn=respond, inputs=[msg, chatbot, topic, memory_flag], outputs=[msg, chatbot])
     clear.click(lambda: [], None, chatbot, queue=False)
     clear.click(lambda: "", None, msg, queue=False)
 
-    default_starters = get_examples_for_topic("Daily Life")
+    default_starters = get_examples_for_topic("Daily Life", "intermediate")
     demo.load(fn=lambda: tuple(default_starters[:3]), outputs=starter_buttons, queue=False)
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)