Update app.py

app.py

@@ -52,6 +52,45 @@ def get_weather(location: str) -> str:
 
 
 
+from smolagents import tool
+from smolagents.llms import HfApiModel
+
+@tool
+def create_personal_schedule(tasks: list, focus_hours: str, restrictions: str, duration_scope: str = "day") -> str:
+    """A tool that generates a personalized daily or weekly schedule using an LLM.
+    
+    Args:
+        tasks: A list of tasks (e.g., ["write report", "book flight", "team meeting"])
+        focus_hours: User-defined focus hours (e.g., "10:00-13:00, 16:00-18:00")
+        restrictions: Any restrictions like "no meetings after 7 PM"
+        duration_scope: Either "day" or "week"
+    """
+
+    prompt = f"""
+    You are a smart planning assistant. Based on the following details, generate a personalized schedule:
+    
+    Tasks: {tasks}
+    Focus Hours: {focus_hours}
+    Restrictions: {restrictions}
+    Scope: {duration_scope}
+
+    Allocate realistic time durations to tasks (e.g., booking a flight = 30 min, working on project = 3 hrs). 
+    Try to match important tasks with focus hours and respect user restrictions.
+
+    Format the output as a bullet-point schedule.
+    """
+
+    # Call the LLM (you can change the model_id if needed)
+    model = HfApiModel(
+        model_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        temperature=0.5,
+        max_tokens=1024,
+    )
+
+    output = model(prompt)
+    return output
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -73,7 +112,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer ,  get_current_time_in_timezone, get_weather], ## add your tools here (don't remove final answer)
+    tools=[final_answer ,  get_current_time_in_timezone, get_weather , create_personal_schedule], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,