Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import time
 from dotenv import load_dotenv
 import gradio as gr
 from groq import Groq
@@ -7,118 +6,120 @@ from groq import Groq
 load_dotenv()
 
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-if not GROQ_API_KEY:
-    raise ValueError("GROQ_API_KEY not found in .env file or environment variables.")
-
-client = Groq(api_key=GROQ_API_KEY)
-
-def chatbot_response(user_input, action, language, conversation_history):
 
+client = Groq(api_key=GROQ_API_KEY) if GROQ_API_KEY else None
+
+def chatbot_response(user_input, action, language, conversation_history, name, proficiency_level):
+    if not client:
+        error_msg = "⚠️ Error: Missing GROQ_API_KEY. Please check your .env file."
+        conversation_history.append((user_input, error_msg))
+        return conversation_history
+    
+    system_message = f"""You are a {language} tutor for {name or 'a student'} ({proficiency_level} level). 
+    Provide personalized, level-appropriate responses. Use their name occasionally. 
+    For games, maintain context until completion. Acknowledge different dialects where relevant.
+    
+    **Important**: The user is communicating via text - focus on written errors. 
+    Only provide pronunciation help if explicitly requested.
+    
+    When correcting errors:
+    1. Clearly indicate the error with 🚨Error Alret🚨
+    2. Explain why it's incorrect
+    3. Provide the correct version
+    4. Offer a brief practice example"""
+    
     if action == "vocabulary":
-        prompt = f"Act as a {language} language tutor. Provide 5 vocabulary words for a beginner learning {language} for travel."
-        action_text = "**Get Daily Vocabulary**"
+        user_message = f"Provide 5 {language} vocabulary words for {proficiency_level} travelers."
     elif action == "grammar":
-        prompt = f"Act as a {language} language tutor. Provide a mini-lesson on grammar rules (e.g., verb conjugations, sentence structure)."
-        action_text = "**Grammar Lesson**"
+        user_message = f"Explain a {language} grammar concept suitable for {proficiency_level} learners."
     elif action == "conversation":
-        prompt = f"Act as a {language} language tutor. Simulate a real-life conversation (e.g., ordering food, asking for directions)."
-        action_text = "**Practice Conversation**"
+        user_message = f"Start a {language} conversation practice. First offer 3 topic choices and 4th option that offers another 3 options."
     elif action == "game":
-        prompt = f"Act as a {language} language tutor. Create a language game like word matching, fill-in-the-blank, or word puzzles."
-        action_text = "**Play a Language Game**"
+        user_message = f"Create an interactive {language} language game. Maintain game context."
     else:
-        prompt = f"Act as a {language} language tutor. Respond to the following user input: {user_input}"
-        action_text = ""
-
-
-    if action != "chat":
-        conversation_history.append((f"User selected: {action_text}", ""))  
-    elif user_input.strip():  
-        conversation_history.append((user_input, ""))  
+        user_message = user_input
 
+    conversation_history.append((user_message, ""))
+    
     try:
+        messages = [{"role": "system", "content": system_message}]
+        for msg, response in conversation_history:
+            messages.append({"role": "user", "content": msg})
+            if response:
+                messages.append({"role": "assistant", "content": response.replace(f"**{language} Tutor:** ", "")})
+        
         completion = client.chat.completions.create(
-            model="llama-3.3-70b-versatile", 
-            messages=[
-                {"role": "system", "content": "You are a helpful and knowledgeable language tutor."},
-                {"role": "user", "content": prompt}
-            ],
-            temperature=0.8,
-            max_tokens=2000,  
-            stream=True,  
+            model="llama3-70b-8192",
+            messages=messages,
+            temperature=0.7,
+            max_tokens=2000,
+            stream=True,
         )
 
-
         chatbot_message = ""
         for chunk in completion:
             if chunk.choices[0].delta.content:
                 chatbot_message += chunk.choices[0].delta.content
-                if action != "chat":
-                    conversation_history[-1] = (f"User selected: {action_text}", f"**{language} Tutor:** {chatbot_message}")
-                else:
-                    conversation_history[-1] = (user_input, f"**{language} Tutor:** {chatbot_message}")
-                yield conversation_history 
-
+                conversation_history[-1] = (user_message, f"**{language} Tutor:** {chatbot_message}")
+                yield conversation_history
 
     except Exception as e:
-        chatbot_message = f"Sorry, an error occurred: {str(e)}. Please try again later."
-        if action != "chat":
-            conversation_history[-1] = (f"User selected: {action_text}", f"**{language} Tutor:** {chatbot_message}")
-        else:
-            conversation_history[-1] = (user_input, f"**{language} Tutor:** {chatbot_message}")
-        yield conversation_history  
+        error_msg = f"⚠️ Error: {str(e)}. Please try again or reload."
+        conversation_history[-1] = (user_message, f"**{language} Tutor:** {error_msg}")
+        yield conversation_history
 
-        
-with gr.Blocks(theme=gr.themes.Default(primary_hue="green", secondary_hue="gray")) as demo:
-    gr.Markdown("# Multi-Language Learning Buddy")
-    gr.Markdown("Welcome! Practice vocabulary, grammar, conversation, and play language games in multiple languages.")
+def clear_chat():
+    return [], [] 
 
-    language = gr.Dropdown(choices=["French", "Spanish", "German", "Italian"], label="Select Language", value="French")
-
-    chatbot = gr.Chatbot(label="Chat with the Language Learning Buddy")
-    user_input = gr.Textbox(label="Your Message", placeholder="Type your question or response here...", lines=2)
-    send_button = gr.Button("Send")
+with gr.Blocks(theme=gr.themes.Default(primary_hue="green")) as demo:
+    gr.Markdown("# 🌍 Multi-Language Learning Buddy")
+    gr.Markdown("Welcome! Practice vocabulary, grammar, conversation, and play language games in multiple languages.")
 
+    
     with gr.Row():
-        vocab_button = gr.Button("Get Daily Vocabulary")
-        grammar_button = gr.Button("Grammar Lesson")
-        conv_button = gr.Button("Practice Conversation")
-        game_button = gr.Button("Play a Language Game")
-
-    # Store conversation history
+        language = gr.Dropdown(["French", "Spanish", "German", "Italian", "Portuguese"], label="Language", value="French")
+        proficiency = gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="Your Level", value="Beginner")
+        name = gr.Textbox(label="Your Name", placeholder="Enter your name...")
+    
+    chatbot = gr.Chatbot(height=500)
     conversation_history = gr.State([])
-
-    # Define button actions
-    vocab_button.click(
-        fn=chatbot_response,
-        inputs=[user_input, gr.State("vocabulary"), language, conversation_history],
-        outputs=chatbot
-    )
-    grammar_button.click(
-        fn=chatbot_response,
-        inputs=[user_input, gr.State("grammar"), language, conversation_history],
-        outputs=chatbot
-    )
-    conv_button.click(
-        fn=chatbot_response,
-        inputs=[user_input, gr.State("conversation"), language, conversation_history],
-        outputs=chatbot
-    )
-    game_button.click(
-        fn=chatbot_response,
-        inputs=[user_input, gr.State("game"), language, conversation_history],
-        outputs=chatbot
-    )
-
-    send_button.click(
-        fn=chatbot_response,
-        inputs=[user_input, gr.State("chat"), language, conversation_history],
-        outputs=chatbot
-    ).then(
-        lambda: "", 
-        inputs=None,
-        outputs=user_input
+    
+    with gr.Row():
+        user_input = gr.Textbox(placeholder="Type here or choose an activity below...", show_label=False)
+        send_btn = gr.Button("Send", variant="primary")
+    
+    with gr.Row():
+        vocab_btn = gr.Button("📚 Daily Vocabulary")
+        grammar_btn = gr.Button("📖 Grammar Lesson")
+        conv_btn = gr.Button("💬 Practice Conversation")
+        game_btn = gr.Button("🎮 Language Game")
+        clear_btn = gr.Button("🔄 Clear Chat")
+
+    input_components = [user_input, gr.State("chat"), language, conversation_history, name, proficiency]
+    
+    send_btn.click(
+        chatbot_response, inputs=input_components, outputs=chatbot
+    ).then(lambda: "", None, user_input)
+    
+    user_input.submit(
+        chatbot_response, inputs=input_components, outputs=chatbot
+    ).then(lambda: "", None, user_input)
+    
+    action_btns = [vocab_btn, grammar_btn, conv_btn, game_btn]
+    for btn, action in zip(action_btns, ["vocabulary", "grammar", "conversation", "game"]):
+        btn.click(
+            chatbot_response,
+            inputs=[gr.State(""), gr.State(action)] + input_components[2:],
+            outputs=chatbot
+        )
+    
+    clear_btn.click(
+        clear_chat,
+        outputs=[chatbot, conversation_history], 
     )
 
-# Launch the interface
-demo.launch()
+if __name__ == "__main__":
+    if not GROQ_API_KEY:
+        print("❌ Error: GROQ_API_KEY missing. Create a .env file with your key.")
+    else:
+        demo.launch()