Update app.py

added replicate and langchain

app.py

@@ -5,11 +5,11 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# 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")
+# Import replicate directly to generate images using Stable Diffusion.
+try:
+    import replicate
+except ModuleNotFoundError:
+    replicate = None
 
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
@@ -114,7 +114,7 @@ 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 original image_generation_tool.
+    # 3. Generate a promotional image using the original smolagents 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}"
@@ -123,13 +123,18 @@ def prompte_chakwal(custom_note: str = "", num_results: int = 3) -> str:
     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}")
+    # 4. Generate a promotional image using the Replicate API (Stable Diffusion) if available.
+    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}")
+        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)