Rename app.py_orig to app.py

app.py

@@ -0,0 +1,156 @@
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+import datetime
+import requests
+import pytz
+import yaml
+import random
+from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI
+
+@tool
+def excuse_generator(situation: str, formality_level: str = "casual") -> str:
+    """A tool that generates creative excuses for various situations
+    
+    Args:
+        situation: The situation requiring an excuse (e.g., 'late_to_meeting', 'missing_deadline', 'forgot_birthday')
+        formality_level: How formal the excuse should be ('casual', 'professional', or 'elaborate')
+    """
+    
+    excuse_components = {
+        'tech_problems': [
+            "my smart home went into lockdown mode",
+            "my AI assistant started writing poetry instead of setting alarms",
+            "my laptop decided to update all 247 applications at once",
+            "my coffee maker hacked my calendar and deleted all my meetings",
+            "my virtual reality headset trapped me in a meditation session"
+        ],
+        'transportation': [
+            "a parade of ducks was crossing the road",
+            "my car's GPS insisted on taking me through a renaissance fair",
+            "my bicycle was requisitioned by time travelers",
+            "a group of joggers mistook me for their running coach",
+            "my helicopter uber got stuck in a cloud"
+        ],
+        'pet_related': [
+            "my cat unplugged my alarm clock to charge their phone",
+            "my dog ate my keys and the vet was closed",
+            "my goldfish needed emotional support",
+            "my hamster staged a protest for better wheel conditions",
+            "my pet plant needed emergency repotting"
+        ],
+        'weather': [
+            "a localized rain cloud followed me around all morning",
+            "a rainbow appeared and I had to document it for science",
+            "the wind kept pushing me in the wrong direction",
+            "a mysterious fog turned my neighborhood into a maze",
+            "the sun was in my eyes (it was night time)"
+        ],
+        'professional': [
+            "I was trapped in an endless email chain",
+            "the coffee machine initiated a workplace revolution",
+            "my keyboard only typed in emoji",
+            "my desk chair rolled me into an impromptu meeting",
+            "my productivity app became too productive"
+        ]
+    }
+    
+    situation_categories = {
+        'late_to_meeting': ['tech_problems', 'transportation', 'professional'],
+        'missing_deadline': ['tech_problems', 'professional'],
+        'forgot_birthday': ['tech_problems', 'pet_related'],
+        'missed_call': ['tech_problems', 'weather', 'pet_related'],
+        'general': ['tech_problems', 'transportation', 'pet_related', 'weather', 'professional']
+    }
+    
+    formality_modifiers = {
+        'casual': [
+            "You're not gonna believe this, but...",
+            "So, funny story...",
+            "Well...",
+            "Here's the thing...",
+            "Would you believe..."
+        ],
+        'professional': [
+            "I regret to inform you that...",
+            "Due to unforeseen circumstances...",
+            "I must apologize, as...",
+            "Please be advised that...",
+            "I am writing to inform you that..."
+        ],
+        'elaborate': [
+            "In an unprecedented turn of events that defies explanation...",
+            "Through a series of increasingly improbable occurrences...",
+            "In what can only be described as a cosmic convergence of unlikely events...",
+            "After careful consideration of the extraordinary circumstances...",
+            "In a stunning display of Murphy's Law..."
+        ]
+    }
+    
+    # Default to 'general' if situation not found
+    valid_categories = situation_categories.get(situation, situation_categories['general'])
+    
+    # Select random category and excuse from that category
+    category = random.choice(valid_categories)
+    core_excuse = random.choice(excuse_components[category])
+    
+    # Add formality modifier
+    modifier = random.choice(formality_modifiers.get(formality_level, formality_modifiers['casual']))
+    
+    # Construct the full excuse
+    full_excuse = f"{modifier} {core_excuse}"
+    
+    # Add a random consequence for elaborate excuses
+    if formality_level == 'elaborate':
+        consequences = [
+            "This naturally led to a cascade of bureaucratic complications.",
+            "The authorities have been notified, but they were equally puzzled.",
+            "I'm still waiting for the insurance adjuster to stop laughing.",
+            "The local news has already contacted me for an exclusive.",
+            "Scientists are currently studying the phenomenon."
+        ]
+        full_excuse += f" {random.choice(consequences)}"
+    
+    return full_excuse
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
+    
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+
+final_answer = FinalAnswerTool()
+
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+    custom_role_conversions=None,
+)
+
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
+
+agent = CodeAgent(
+    model=model,
+    tools=[get_current_time_in_timezone, excuse_generator, DuckDuckGoSearchTool(), final_answer],
+    max_steps=6,
+    verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
+    prompt_templates=prompt_templates
+)
+
+GradioUI(agent).launch()

app.py_orig

@@ -1,93 +0,0 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import datetime
-import requests
-import pytz
-import yaml
-import random
-from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
-
-@tool
-def magic_8_ball(question: str, asker_name: str) -> str:
-    """A Magic 8 Ball tool that provides fortune-telling responses to yes/no questions
-    
-    Args:
-        question: The question being asked to the Magic 8 Ball
-        asker_name: The name of the person asking the question
-    """
-    responses = [
-        # Positive responses
-        "It is certain.",
-        "It is decidedly so.",
-        "Without a doubt.",
-        "Yes definitely.",
-        "You may rely on it.",
-        "As I see it, yes.",
-        "Most likely.",
-        "Outlook good.",
-        "Yes.",
-        "Signs point to yes.",
-        
-        # Non-committal responses
-        "Reply hazy, try again.",
-        "Ask again later.",
-        "Better not tell you now.",
-        "Cannot predict now.",
-        "Concentrate and ask again.",
-        
-        # Negative responses
-        "Don't count on it.",
-        "My reply is no.",
-        "My sources say no.",
-        "Outlook not so good.",
-        "Very doubtful."
-    ]
-    
-    # Get random response
-    answer = random.choice(responses)
-    return f"🎱 Dear {asker_name}, you asked: '{question}'\nThe Magic 8 Ball says: {answer}"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-final_answer = FinalAnswerTool()
-
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
-    custom_role_conversions=None,
-)
-
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-agent = CodeAgent(
-    model=model,
-    tools=[get_current_time_in_timezone, magic_8_ball, DuckDuckGoSearchTool(), final_answer],
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
-)
-
-GradioUI(agent).launch()