Update app.py

added langchain

app.py

@@ -5,7 +5,11 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Existing tools...
+# Import LangChain’s Replicate tool for image generation.
+from langchain.tools.replicate import ReplicateTool
+
+# Instantiate LangChain image generation tool using a Stable Diffusion model.
+langchain_image_tool = ReplicateTool(model="stability-ai/stable-diffusion")
 
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
@@ -68,19 +72,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-# Create the image generation tool from the hub.
+# Create the image generation tool from smolagents.
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 # ── NEW TOOL: prompte_chakwal ──
 @tool
 def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
     """
-    A tool that promotes Chakwal by retrieving real-time information and generating a promotional image.
-    It provides the current local time, live news about Chakwal from DuckDuckGo, and creates an image that
-    highlights the city's natural beauty and vibrant culture.
+    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.
     
     Args:
-        custom_note: A custom note to add a specific theme to the generated image.
+        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.
     """
     output_lines = []
@@ -110,15 +114,23 @@ def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
     else:
         output_lines.append("\nNo real-time information available.")
     
-    # 3. Generate a promotional image using the image generation tool
+    # 3. Generate a promotional image using the original image_generation_tool.
     image_prompt = "A breathtaking view of Chakwal showcasing its natural beauty, vibrant culture, and historical landmarks"
     if custom_note:
         image_prompt += f". Theme: {custom_note}"
         
     generated_image = image_generation_tool(image_prompt)
-    output_lines.append("\nGenerated Promotional Image:")
+    output_lines.append("\nGenerated Promotional Image (smolagents):")
     output_lines.append(f"{generated_image}")
     
+    # 4. Generate a promotional image using LangChain's image generation (ReplicateTool)
+    try:
+        generated_image_langchain = langchain_image_tool.run(image_prompt)
+        output_lines.append("\nGenerated Promotional Image (LangChain):")
+        output_lines.append(f"{generated_image_langchain}")
+    except Exception as e:
+        output_lines.append(f"\nError generating image with LangChain: {e}")
+    
     return "\n".join(output_lines)