Update app.py

app.py

@@ -5,6 +5,9 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# Import AgentText to return a composite Markdown message
+from smolagents.agent_types import AgentText
+
 # Import replicate directly to generate images using Stable Diffusion.
 try:
     import replicate
@@ -16,7 +19,7 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     """
     A tool that shares interesting facts about Chakwal, highlighting its flora, fauna, natural landscapes,
     and cultural heritage. It also shows the current local time in Chakwal (Asia/Karachi timezone).
-    
+
     Args:
         arg1: A custom note or additional message to include in the output.
         arg2: The number of detailed facts to include about Chakwal (1 to 4).
@@ -60,7 +63,7 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
 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').
     """
@@ -77,12 +80,14 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 # ── NEW TOOL: prompte_chakwal ──
 @tool
-def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
+def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> AgentText:
     """
     A tool that promotes Chakwal by retrieving real-time information and generating promotional images.
     It provides the current local time, live news about Chakwal from DuckDuckGo, and creates images that
-    highlight the city's natural beauty and vibrant culture using two different image generators.
-    
+    highlight the city's natural beauty and vibrant culture.
+
+    The final output is a composite Markdown message that includes text details and an embedded image.
+
     Args:
         custom_note: A custom note to add a specific theme to the generated images.
         num_results: The maximum number of real-time search results (news/information) to include.
@@ -93,7 +98,7 @@ def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
     try:
         tz = pytz.timezone("Asia/Karachi")
         current_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        output_lines.append(f"Current local time in Chakwal: {current_time}")
+        output_lines.append(f"**Current local time in Chakwal:** {current_time}")
     except Exception as e:
         output_lines.append("Error fetching current time in Chakwal.")
     
@@ -103,13 +108,13 @@ def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
     search_results = duck_tool.run(query)
     
     if isinstance(search_results, list):
-        output_lines.append("\nReal-Time News and Information:")
+        output_lines.append("\n**Real-Time News and Information:**")
         for i, result in enumerate(search_results[:num_results]):
             title = result.get("title", "No Title")
             snippet = result.get("snippet", "No description available")
-            output_lines.append(f"{i+1}. {title}: {snippet}")
+            output_lines.append(f"{i+1}. **{title}**: {snippet}")
     elif isinstance(search_results, str):
-        output_lines.append("\nReal-Time News and Information:")
+        output_lines.append("\n**Real-Time News and Information:**")
         output_lines.append(search_results)
     else:
         output_lines.append("\nNo real-time information available.")
@@ -119,24 +124,35 @@ def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
     if custom_note:
         image_prompt += f". Theme: {custom_note}"
         
-    generated_image = image_generation_tool(image_prompt)
-    output_lines.append("\nGenerated Promotional Image (smolagents):")
-    output_lines.append(f"{generated_image}")
+    generated_image_url = image_generation_tool(image_prompt)
+    output_lines.append("\n**Generated Promotional Image (smolagents):**")
+    output_lines.append(f"Image URL: {generated_image_url}")
     
     # 4. Generate a promotional image using the Replicate API (Stable Diffusion) if available.
+    generated_image_replicate = None
     if replicate:
         try:
             stable_diffusion = replicate.models.get("stability-ai/stable-diffusion")
             # The predict method typically returns a list of image URLs.
-            generated_image_replicate = stable_diffusion.predict(prompt=image_prompt)
-            output_lines.append("\nGenerated Promotional Image (Replicate):")
-            output_lines.append(f"{generated_image_replicate}")
+            result = stable_diffusion.predict(prompt=image_prompt)
+            if isinstance(result, list) and len(result) > 0:
+                generated_image_replicate = result[0]
+            else:
+                generated_image_replicate = result
+            output_lines.append("\n**Generated Promotional Image (Replicate):**")
+            output_lines.append(f"Image URL: {generated_image_replicate}")
         except Exception as e:
             output_lines.append(f"\nError generating image with Replicate: {e}")
     else:
         output_lines.append("\nReplicate module is not available. Install it via 'pip install replicate'.")
     
-    return "\n".join(output_lines)
+    # Choose one image to embed: Prefer the Replicate image if available, else the smolagents image.
+    final_image_url = generated_image_replicate if generated_image_replicate else generated_image_url
+    
+    # Create a composite Markdown with text details and embed the chosen image.
+    composite_markdown = "\n".join(output_lines) + f"\n\n![Chakwal Promotional Image]({final_image_url})"
+    
+    return AgentText(composite_markdown)
 
 
 # Set up the final answer tool.