update tools

app.py

@@ -35,20 +35,25 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def generate_post(subject: str, max_length: int = 200) -> str:
-    """Generates a well-structured post about a given subject with a length limit.
+    """Generates a concise and well-structured post about a given subject.
     
     Args:
         subject: The topic to generate a post about.
         max_length: The maximum length of the post in characters (default: 200).
     
     Returns:
-        A concise, engaging post about the subject.
+        A well-structured, engaging post about the subject.
     """
-    prompt = f"Write a concise and engaging social media post about {subject}. Keep it within {max_length} characters."
-    
-    response = model.generate_text(prompt, max_tokens=max_length // 4)  # Approx. 4 chars per token
-    
-    return response[:max_length]  # Ensure it doesn't exceed max_length
+    prompt = f"Write a concise and engaging social media post about {subject}. Limit the response to {max_length} characters."
+
+    try:
+        global model
+        response = model.generate_text(prompt) 
+        
+        return response[:max_length]
+    except Exception as e:
+        return f"Error generating post: {str(e)}"
+
 
 
 final_answer = FinalAnswerTool()