Update app.py

app.py

@@ -1,9 +1,22 @@
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent, tool
 import yaml
 from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
 import random
 
+# Mock Model that works without external API calls
+class LocalModel:
+    def __init__(self):
+        self.max_tokens = 2096
+        self.temperature = 0.7
+        self.model_id = 'local-model'
+    
+    def generate(self, messages, **kwargs):
+        # Simple response generation without external API
+        class Response:
+            def __init__(self, content):
+                self.content = content
+        return Response("Using local processing for English tutoring.")
+
 @tool
 def explain_word(word: str) -> str:
     """Explains an English word with definition, pronunciation, examples, and usage tips.
@@ -14,7 +27,7 @@ def explain_word(word: str) -> str:
     Returns:
         A comprehensive explanation of the word
     """
-    # Simple word database for demonstration
+    # Enhanced word database with common English words
     word_database = {
         "hello": {
             "definition": "A greeting used when meeting someone or starting a conversation",
@@ -51,10 +64,46 @@ def explain_word(word: str) -> str:
             ],
             "synonyms": ["study", "acquire", "master", "understand"],
             "usage_tip": "Often followed by 'to' + verb (learn to speak) or 'about' + noun (learn about history)."
+        },
+        "great": {
+            "definition": "Of an extent, amount, or intensity considerably above average; excellent",
+            "pronunciation": "grayt",
+            "part_of_speech": "adjective",
+            "examples": [
+                "She did a great job on the project.",
+                "The Great Wall of China is amazing.",
+                "Great! Let's start the lesson."
+            ],
+            "synonyms": ["excellent", "wonderful", "fantastic", "outstanding"],
+            "usage_tip": "Can be used to describe size, quality, or as an exclamation of approval."
+        },
+        "friend": {
+            "definition": "A person you know well and regard with affection and trust",
+            "pronunciation": "frend",
+            "part_of_speech": "noun",
+            "examples": [
+                "She is my best friend.",
+                "I made a new friend at school.",
+                "Friends help each other."
+            ],
+            "synonyms": ["buddy", "pal", "companion", "mate"],
+            "usage_tip": "Friends are people you choose to spend time with and share experiences."
+        },
+        "happy": {
+            "definition": "Feeling or showing pleasure or contentment",
+            "pronunciation": "HAP-ee",
+            "part_of_speech": "adjective",
+            "examples": [
+                "I'm happy to see you!",
+                "The children look happy playing.",
+                "Happy birthday to you!"
+            ],
+            "synonyms": ["joyful", "cheerful", "delighted", "glad"],
+            "usage_tip": "Used to describe positive emotions and feelings of joy."
         }
     }
     
-    word_lower = word.lower()
+    word_lower = word.lower().strip()
     
     if word_lower in word_database:
         info = word_database[word_lower]
@@ -69,14 +118,14 @@ def explain_word(word: str) -> str:
         response += f"**Usage Tip:** {info['usage_tip']}"
         return response
     else:
-        return f"I don't have detailed information about '{word}' in my database. However, here's a general explanation:\n\n'{word}' is an English word. To learn more about it, try:\n- Looking it up in a dictionary\n- Using it in sentences\n- Finding synonyms and antonyms\n- Practicing pronunciation\n\nWould you like me to help with grammar rules or suggest other vocabulary words to learn?"
+        return f"## Word: {word.capitalize()}\n\nI don't have detailed information about '{word}' in my current database. Here are some suggestions:\n\n**How to learn about new words:**\n- Look it up in a dictionary\n- Use it in sentences\n- Find synonyms and antonyms\n- Practice pronunciation\n- Read it in context\n\n**Available words in my database:**\nhello, beautiful, learn, great, friend, happy\n\nWould you like me to explain one of these words or help with grammar lessons?"
 
 @tool
 def grammar_lesson(topic: str) -> str:
     """Provides grammar lessons on various English topics.
     
     Args:
-        topic: The grammar topic to explain (tenses, articles, prepositions, etc.)
+        topic: The grammar topic to explain (present_tense, articles, prepositions, etc.)
     
     Returns:
         A detailed grammar lesson with examples
@@ -103,288 +152,4 @@ def grammar_lesson(topic: str) -> str:
         "articles": {
             "title": "Articles (a, an, the)",
             "explanation": "Articles are words that introduce nouns. English has three articles: 'a', 'an', and 'the'.",
-            "types": [
-                {
-                    "name": "Indefinite Articles (a, an)",
-                    "form": "'a' before consonant sounds, 'an' before vowel sounds",
-                    "examples": ["a book", "an apple", "a university", "an hour"],
-                    "usage": "For non-specific, countable singular nouns"
-                },
-                {
-                    "name": "Definite Article (the)",
-                    "form": "'the' before specific nouns",
-                    "examples": ["the book I'm reading", "the sun", "the United States"],
-                    "usage": "For specific nouns, unique things, or previously mentioned items"
-                }
-            ]
-        },
-        "prepositions": {
-            "title": "Prepositions (in, on, at, etc.)",
-            "explanation": "Prepositions show relationships between words, often indicating time, place, or direction.",
-            "types": [
-                {
-                    "name": "Prepositions of Time",
-                    "form": "in, on, at",
-                    "examples": ["in 2023", "on Monday", "at 3 o'clock", "in the morning"],
-                    "usage": "in (months/years), on (days/dates), at (specific times)"
-                },
-                {
-                    "name": "Prepositions of Place",
-                    "form": "in, on, at, under, above",
-                    "examples": ["in the house", "on the table", "at school", "under the bed"],
-                    "usage": "Describe where something is located"
-                }
-            ]
-        }
-    }
-    
-    topic_lower = topic.lower().replace(" ", "_")
-    
-    if topic_lower in grammar_lessons:
-        lesson = grammar_lessons[topic_lower]
-        response = f"## {lesson['title']}\n\n"
-        response += f"{lesson['explanation']}\n\n"
-        
-        for grammar_type in lesson['types']:
-            response += f"### {grammar_type['name']}\n"
-            response += f"**Form:** {grammar_type['form']}\n"
-            response += f"**Usage:** {grammar_type['usage']}\n"
-            response += f"**Examples:**\n"
-            for example in grammar_type['examples']:
-                response += f"- {example}\n"
-            response += "\n"
-        
-        return response
-    else:
-        available_topics = list(grammar_lessons.keys())
-        return f"I don't have a lesson on '{topic}'. Here are available grammar topics:\n\n" + \
-               "\n".join([f"- {topic.replace('_', ' ').title()}" for topic in available_topics]) + \
-               "\n\nPlease choose one of these topics for a detailed lesson."
-
-@tool
-def practice_exercise(exercise_type: str) -> str:
-    """Generates practice exercises for English learning.
-    
-    Args:
-        exercise_type: Type of exercise (vocabulary, grammar, sentence_building)
-    
-    Returns:
-        A practice exercise with questions
-    """
-    exercises = {
-        "vocabulary": [
-            {
-                "question": "Choose the correct word to complete the sentence:",
-                "sentence": "The weather is very _____ today.",
-                "options": ["beautiful", "beautifully", "beauty", "beautify"],
-                "correct": "beautiful",
-                "explanation": "'Beautiful' is an adjective that describes the noun 'weather'."
-            },
-            {
-                "question": "What is the opposite of 'difficult'?",
-                "options": ["easy", "hard", "complex", "challenging"],
-                "correct": "easy",
-                "explanation": "'Easy' is the antonym of 'difficult'."
-            }
-        ],
-        "grammar": [
-            {
-                "question": "Choose the correct article:",
-                "sentence": "I saw _____ elephant at the zoo.",
-                "options": ["a", "an", "the", "no article"],
-                "correct": "an",
-                "explanation": "Use 'an' before words starting with vowel sounds. 'Elephant' starts with 'e'."
-            },
-            {
-                "question": "Select the correct tense:",
-                "sentence": "She _____ to school every day.",
-                "options": ["go", "goes", "going", "went"],
-                "correct": "goes",
-                "explanation": "For third person singular in simple present, add 's' to the verb."
-            }
-        ],
-        "sentence_building": [
-            {
-                "question": "Arrange these words to make a correct sentence:",
-                "words": ["I", "every", "morning", "coffee", "drink"],
-                "correct": "I drink coffee every morning.",
-                "explanation": "Subject + verb + object + time expression is a common sentence pattern."
-            },
-            {
-                "question": "Complete the sentence with the correct word order:",
-                "sentence": "Yesterday, [beautiful/very/a/saw/I/movie]",
-                "correct": "Yesterday, I saw a very beautiful movie.",
-                "explanation": "Time + subject + verb + article + adverb + adjective + noun"
-            }
-        ]
-    }
-    
-    if exercise_type in exercises:
-        exercise = random.choice(exercises[exercise_type])
-        response = f"## {exercise_type.title()} Practice\n\n"
-        response += f"**{exercise['question']}**\n\n"
-        
-        if 'sentence' in exercise:
-            response += f"Sentence: {exercise['sentence']}\n\n"
-        
-        if 'words' in exercise:
-            response += f"Words to arrange: {', '.join(exercise['words'])}\n\n"
-        
-        if 'options' in exercise:
-            response += "Options:\n"
-            for i, option in enumerate(exercise['options'], 1):
-                response += f"{i}. {option}\n"
-            response += "\n"
-        
-        response += f"**Correct Answer:** {exercise['correct']}\n\n"
-        response += f"**Explanation:** {exercise['explanation']}"
-        
-        return response
-    else:
-        return f"Available exercise types: vocabulary, grammar, sentence_building\n\nPlease choose one of these types for practice."
-
-# Setup agent
-final_answer = FinalAnswerTool()
-
-# Using Mistral model for better English instruction
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.7,
-    model_id='mistralai/Mistral-7B-Instruct-v0.1',
-    custom_role_conversions={
-        "system": "<s>[INST] <<SYS>>\n{message}\n<</SYS>>\n\n",
-        "human": "{message} [/INST]",
-        "assistant": "{message} </s><s>[INST] "
-    },
-)
-
-# Define all required prompt templates
-prompt_templates = {
-    "react-system": """You are an English tutor AI that helps people improve their English skills.
-You have access to the following tools:
-
-{tools_string}
-
-You should help users with:
-1. Explaining English words (definitions, pronunciation, examples)
-2. Teaching grammar rules and concepts
-3. Providing practice exercises
-
-Always be encouraging and patient. Explain things clearly with examples.
-
-To use a tool, output an action in this format:
-```py
-action = tool_name(arg1=value1, arg2=value2, ...)
-```
-
-Always use Python syntax for calling tools. After using a tool, provide additional context or explanation to help the user learn better. Use final_answer to conclude your response.
-""",
-    "react-human": "{input_string}",
-    "react-assistant": "{agent_response}",
-    "final_answer": {
-        "pre_messages": "Here's your English lesson:",
-        "template": "Final Answer: {answer}",
-        "post_messages": "Keep practicing! Would you like to learn something else?"
-    },
-    "system_prompt": "You are a helpful English tutor that explains words, teaches grammar, and provides practice exercises.",
-    "planning": {
-        "pre_messages": "I'll help you improve your English skills.",
-        "template": "Plan: {planning}",
-        "post_messages": "Let me provide you with the best English learning experience.",
-        "initial_plan": "I'll identify what the user wants to learn about English and use the appropriate tools to help them.",
-        "update_plan_pre_messages": "Let me adjust my teaching approach.",
-        "update_plan_template": "Updated Plan: {planning}",
-        "update_plan_post_messages": "Now I'll continue with this updated approach."
-    },
-    "managed_agent": {
-        "pre_messages": "Let me help you with your English learning.",
-        "template": "{agent_response}",
-        "post_messages": "I hope this helps improve your English!",
-        "task": "Help users improve their English skills through word explanations, grammar lessons, and practice exercises.",
-        "report": "English learning session completed with explanations and practice exercises provided."
-    }
-}
-
-# Create the agent with all required templates
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer, explain_word, grammar_lesson, practice_exercise],
-    max_steps=5,
-    verbosity_level=1,
-    prompt_templates=prompt_templates
-)
-
-# Create a custom Gradio UI for English learning
-class EnglishTutorUI(GradioUI):
-    def launch(self, **kwargs):
-        import gradio as gr
-        
-        with gr.Blocks(theme=gr.themes.Soft(), title="English Tutor AI") as demo:
-            gr.Markdown("# 🇬🇧 English Tutor AI")
-            gr.Markdown("Improve your English with AI assistance!")
-            
-            with gr.Tab("Word Explanation"):
-                with gr.Row():
-                    word_input = gr.Textbox(
-                        label="Enter an English word",
-                        placeholder="e.g., beautiful, learn, hello"
-                    )
-                    word_btn = gr.Button("Explain Word", variant="primary")
-                word_output = gr.Markdown()
-                
-                word_btn.click(
-                    lambda word: agent.run(f"Explain the word '{word}' with definition, pronunciation, examples, and usage tips"),
-                    inputs=[word_input],
-                    outputs=[word_output]
-                )
-            
-            with gr.Tab("Grammar Lessons"):
-                with gr.Row():
-                    grammar_topic = gr.Radio(
-                        ["Present Tense", "Articles", "Prepositions"],
-                        label="Choose a grammar topic"
-                    )
-                    grammar_btn = gr.Button("Get Lesson", variant="primary")
-                grammar_output = gr.Markdown()
-                
-                grammar_btn.click(
-                    lambda topic: agent.run(f"Give me a grammar lesson on {topic}"),
-                    inputs=[grammar_topic],
-                    outputs=[grammar_output]
-                )
-            
-            with gr.Tab("Practice Exercises"):
-                with gr.Row():
-                    exercise_type = gr.Radio(
-                        ["Vocabulary", "Grammar", "Sentence Building"],
-                        label="Choose exercise type"
-                    )
-                    exercise_btn = gr.Button("Get Exercise", variant="primary")
-                exercise_output = gr.Markdown()
-                
-                exercise_btn.click(
-                    lambda ex_type: agent.run(f"Give me a {ex_type.lower()} practice exercise"),
-                    inputs=[exercise_type],
-                    outputs=[exercise_output]
-                )
-            
-            with gr.Tab("Free Chat"):
-                with gr.Row():
-                    chat_input = gr.Textbox(
-                        label="Ask anything about English",
-                        placeholder="e.g., What's the difference between 'a' and 'an'?"
-                    )
-                    chat_btn = gr.Button("Ask", variant="primary")
-                chat_output = gr.Markdown()
-                
-                chat_btn.click(
-                    lambda question: agent.run(question),
-                    inputs=[chat_input],
-                    outputs=[chat_output]
-                )
-        
-        demo.launch(**kwargs)
-
-# Launch the English Tutor
-if __name__ == "__main__":
-    EnglishTutorUI(agent).launch()
+            "types": [