Update app.py

app.py

@@ -1,60 +1,131 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import datetime
-import requests
-import pytz
+from smolagents import CodeAgent, HfApiModel, tool
 import yaml
-import re
+import random
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
-
 @tool
-def generate_passover_coloring_pages(age: int) -> str:
+def generate_famous_quote(topic: str) -> str:
     """
-    Generates Passover-themed coloring pages based on a child's age.
-
+    Generates a famous quote related to the provided topic.
     Args:
-        age (int): The age of the child (1-12).
-
+        topic (str): The topic or theme for the quote.
     Returns:
-        str: A list of suggested Passover-themed coloring pages.
+        str: A famous quote related to the topic and its author.
     """
-
-    if not isinstance(age, int) or not (1 <= age <= 12):
-        return "Error: Please provide a valid age between 1 and 12."
-
-    complexity_levels = {
-        (1, 3): "simple",
-        (4, 6): "basic",
-        (7, 9): "intermediate",
-        (10, 12): "advanced"
+    # Dictionary of quotes by topic
+    quotes_by_topic = {
+        "love": [
+            "The greatest happiness of life is the conviction that we are loved; loved for ourselves, or rather, loved in spite of ourselves. — Victor Hugo",
+            "Love is composed of a single soul inhabiting two bodies. — Aristotle",
+            "Being deeply loved by someone gives you strength, while loving someone deeply gives you courage. — Lao Tzu",
+            "Where there is love, there is life. — Mahatma Gandhi",
+            "The best thing to hold onto in life is each other. — Audrey Hepburn"
+        ],
+        "wisdom": [
+            "The only true wisdom is in knowing you know nothing. — Socrates",
+            "Wisdom is not a product of schooling but of the lifelong attempt to acquire it. — Albert Einstein",
+            "The fool doth think he is wise, but the wise man knows himself to be a fool. — William Shakespeare",
+            "By three methods we may learn wisdom: First, by reflection, which is noblest; Second, by imitation, which is easiest; and third by experience, which is the bitterest. — Confucius",
+            "The beginning of wisdom is the definition of terms. — Socrates"
+        ],
+        "success": [
+            "Success is not final, failure is not fatal: It is the courage to continue that counts. — Winston Churchill",
+            "The way to get started is to quit talking and begin doing. — Walt Disney",
+            "Don't be distracted by criticism. Remember, the only taste of success some people get is to take a bite out of you. — Zig Ziglar",
+            "Success usually comes to those who are too busy to be looking for it. — Henry David Thoreau",
+            "I never dreamed about success, I worked for it. — Estée Lauder"
+        ],
+        "happiness": [
+            "Happiness is not something ready-made. It comes from your own actions. — Dalai Lama",
+            "The happiness of your life depends upon the quality of your thoughts. — Marcus Aurelius",
+            "The most important thing is to enjoy your life—to be happy—it's all that matters. — Audrey Hepburn",
+            "Happiness is when what you think, what you say, and what you do are in harmony. — Mahatma Gandhi",
+            "Count your age by friends, not years. Count your life by smiles, not tears. — John Lennon"
+        ],
+        "time": [
+            "The two most powerful warriors are patience and time. — Leo Tolstoy",
+            "Time is what we want most, but what we use worst. — William Penn",
+            "Yesterday is gone. Tomorrow has not yet come. We have only today. Let us begin. — Mother Teresa",
+            "Time you enjoy wasting is not wasted time. — Marthe Troly-Curtin",
+            "The future starts today, not tomorrow. — Pope John Paul II"
+        ],
+        "friendship": [
+            "A real friend is one who walks in when the rest of the world walks out. — Walter Winchell",
+            "Friendship is born at that moment when one person says to another, 'What! You too? I thought I was the only one.' — C.S. Lewis",
+            "A friend is someone who knows all about you and still loves you. — Elbert Hubbard",
+            "A single rose can be my garden... a single friend, my world. — Leo Buscaglia",
+            "Friendship is the only cement that will ever hold the world together. — Woodrow Wilson"
+        ],
+        "courage": [
+            "Courage is resistance to fear, mastery of fear, not absence of fear. — Mark Twain",
+            "It takes courage to grow up and become who you really are. — E.E. Cummings",
+            "Success is not final, failure is not fatal: It is the courage to continue that counts. — Winston Churchill",
+            "Courage is found in unlikely places. — J.R.R. Tolkien",
+            "You cannot swim for new horizons until you have courage to lose sight of the shore. ��� William Faulkner"
+        ],
+        "change": [
+            "The only way to make sense out of change is to plunge into it, move with it, and join the dance. — Alan Watts",
+            "Change is the law of life. And those who look only to the past or present are certain to miss the future. — John F. Kennedy",
+            "Everyone thinks of changing the world, but no one thinks of changing himself. — Leo Tolstoy",
+            "Be the change that you wish to see in the world. — Mahatma Gandhi",
+            "Those who cannot change their minds cannot change anything. — George Bernard Shaw"
+        ],
+        "education": [
+            "Education is the most powerful weapon which you can use to change the world. — Nelson Mandela",
+            "The function of education is to teach one to think intensively and to think critically. — Martin Luther King, Jr.",
+            "Education is not the filling of a pail, but the lighting of a fire. — W.B. Yeats",
+            "The roots of education are bitter, but the fruit is sweet. — Aristotle",
+            "An investment in knowledge pays the best interest. — Benjamin Franklin"
+        ],
+        "faith": [
+            "Faith is taking the first step even when you don't see the whole staircase. — Martin Luther King, Jr.",
+            "Faith consists in believing when it is beyond the power of reason to believe. — Voltaire",
+            "Keep your faith in all beautiful things; in the sun when it is hidden, in the Spring when it is gone. — Roy R. Gilson",
+            "Faith is the bird that feels the light when the dawn is still dark. — Rabindranath Tagore",
+            "Faith is to believe what you do not see; the reward of this faith is to see what you believe. — Saint Augustine"
+        ],
+        "hope": [
+            "Hope is being able to see that there is light despite all of the darkness. — Desmond Tutu",
+            "We must accept finite disappointment, but never lose infinite hope. — Martin Luther King, Jr.",
+            "Hope is the thing with feathers that perches in the soul. — Emily Dickinson",
+            "Where there is no vision, there is no hope. — George Washington Carver",
+            "A leader is a dealer in hope. — Napoleon Bonaparte"
+        ],
+        "perseverance": [
+            "It does not matter how slowly you go as long as you do not stop. — Confucius",
+            "Perseverance is not a long race; it is many short races one after the other. — Walter Elliot",
+            "Many of life's failures are people who did not realize how close they were to success when they gave up. — Thomas Edison",
+            "Through perseverance many people win success out of what seemed destined to be certain failure. — Benjamin Disraeli",
+            "The difference between a successful person and others is not a lack of strength, not a lack of knowledge, but rather a lack in will. — Vince Lombardi"
+        ]
     }
-    complexity = next(value for (low, high), value in complexity_levels.items() if low <= age <= high)
-
-    passover_pages = {
-        "simple": ["Simple matzah", "Basic Seder plate", "Moses in a basket", "Simple frog (plague)"],
-        "basic": ["Parting of the Red Sea", "Ten Plagues scenes", "Four Questions illustrations", "Afikoman hunt"],
-        "intermediate": ["Detailed Seder table", "Exodus story scenes", "Hebrew letter decorations", "Egyptian pyramids"],
-        "advanced": ["Intricate Haggadah illustrations", "Detailed symbolism scene", "Combined Passover story panels", "Hebrew calligraphy art"]
-    }
-
-    selected_pages = passover_pages[complexity]
-
-    # Попытка сгенерировать изображения
-    try:
-        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-        images = [image_generation_tool(page) for page in selected_pages]
-        return final_answer(images)
-    except Exception as e:
-        return f"Error generating images: {str(e)}"
-
-    response = f"Passover Coloring Pages for a {age}-year-old:\n\n"
-    response += "\n".join(f"- {page}" for page in selected_pages)
     
-    return response
+    # Normalize the topic (convert to lowercase)
+    topic_lower = topic.lower()
+    
+    # Find the closest matching topic
+    if topic_lower in quotes_by_topic:
+        # Topic directly matches
+        relevant_quotes = quotes_by_topic[topic_lower]
+    else:
+        # Search for partial matches in topic keywords
+        matched_topics = []
+        for key in quotes_by_topic:
+            if key in topic_lower or topic_lower in key:
+                matched_topics.append(key)
+        
+        if matched_topics:
+            # Use quotes from the first matching topic
+            relevant_quotes = quotes_by_topic[matched_topics[0]]
+        else:
+            # No match found - use a generic response
+            return f"No famous quotes found for '{topic}'. Try topics like: love, wisdom, success, happiness, time, friendship, courage, change, education, faith, hope, or perseverance."
+    
+    # Return a random quote from the relevant ones
+    return random.choice(relevant_quotes)
 
-# Настройка агента
+# Setup agent
 final_answer = FinalAnswerTool()
 
 model = HfApiModel(
@@ -63,16 +134,75 @@ model = HfApiModel(
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
 )
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-    
+# Custom prompt templates for the quote generator agent
+quote_prompt_templates = {
+    "react-system": """You are a helpful agent that provides famous quotes on various topics.
+You have access to the following tools:
+
+{tools_string}
+
+When a user enters a topic, use the generate_famous_quote tool to find a relevant quote on that topic.
+Return the quote to the user using the final_answer tool.
+
+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, analyze the result to determine what to do next. Use final_answer to return your answer when complete.
+""",
+    "react-human": "{input_string}",
+    "react-assistant": "{agent_response}",
+}
+
+# Load prompts.yaml if it exists, otherwise use our custom templates
+try:
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
+        prompt_templates.update(quote_prompt_templates)
+except:
+    prompt_templates = quote_prompt_templates
+
+# Create the agent
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, generate_passover_coloring_pages],
-    max_steps=6,
+    tools=[final_answer, generate_famous_quote],
+    max_steps=3,  # Reduced steps since this is a simple task
     verbosity_level=1,
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
-
+# Create a custom Gradio UI
+class QuoteGeneratorUI(GradioUI):
+    def launch(self, **kwargs):
+        import gradio as gr
+        
+        with gr.Blocks(theme=gr.themes.Soft()) as demo:
+            gr.Markdown("# Famous Quotes Generator")
+            gr.Markdown("Enter a topic to receive a famous quote related to it.")
+            
+            with gr.Row():
+                with gr.Column():
+                    topic_input = gr.Textbox(label="Enter a topic (e.g., love, wisdom, success)")
+                    submit_btn = gr.Button("Get Quote", variant="primary")
+            
+            with gr.Row():
+                with gr.Column():
+                    quote_output = gr.Markdown(label="Your Quote")
+            
+            def get_quote(topic):
+                if not topic or topic.strip() == "":
+                    return "Please enter a topic to receive a quote."
+                
+                try:
+                    quote = generate_famous_quote(topic)
+                    return f"### Quote on '{topic}':\n\n> {quote}"
+                except Exception as e:
+                    return f"Error generating quote: {str(e)}"
+            
+            submit_btn.click(get_quote, inputs=topic_input, outputs=quote_output)
+        
+        demo.launch(**kwargs)
+
+# Launch the custom UI
+QuoteGeneratorUI(agent).launch()