Update app.py

app.py

@@ -4,10 +4,13 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from transformers import pipeline
 import gradio as gr
+from transformers import pipeline
+
+# Initialize the summarization pipeline globally to avoid reloading it on every call.
+summarizer_pipeline = pipeline("summarization")
 
-# 0. Tool to get the topic from the user (not used directly in the UI)
+# 0. Tool to get the topic from the user (not used directly in the Gradio UI)
 @tool
 def get_topic() -> str:
     """
@@ -28,7 +31,8 @@ def search_web(query: str) -> str:
     Returns:
         str: A snippet from the top search result (excluding Hugging Face-related pages).
     """
-    results = DuckDuckGoSearchTool.search(query=query)
+    duck_tool = DuckDuckGoSearchTool()  # instantiate the search tool
+    results = duck_tool(query=query)
     print("DuckDuckGo Search Results:", results)  # Debug logging
     if results and len(results) > 0:
         # Iterate over results to skip pages with 'huggingface' in the URL
@@ -52,8 +56,13 @@ def summarise_content(text: str) -> str:
     Returns:
         str: The summarized text.
     """
-    summarizer = pipeline("summarization")
-    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
+    # If text is too short, return it directly.
+    if len(text.split()) < 30:
+        return text
+    # If text is too long, trim to the first 1000 characters.
+    if len(text) > 1000:
+        text = text[:1000]
+    summary = summarizer_pipeline(text, max_length=130, min_length=30, do_sample=False)
     return summary[0]['summary_text']
 
 # 3. Tool to generate a social media post based on the summary
@@ -97,19 +106,21 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Custom function to chain the agent actions:
-# 1. Accept the topic
-# 2. Perform a web search
-# 3. Summarize the search result
-# 4. Generate a social media post
+# Function to chain the agent actions:
+# 1. Accept the topic.
+# 2. Use the topic to perform a web search.
+# 3. Summarize the search result.
+# 4. Generate a social media post.
 def agent_interface(user_input_topic: str) -> str:
     topic = user_input_topic
     search_result = search_web(topic)
+    if not search_result.strip():
+        return "No valid search result found."
     summary = summarise_content(search_result)
     post = write_social_media_post(summary)
     return post
 
-# Define a Gradio Interface with a customized landing page
+# Define a Gradio Interface with a customized landing page.
 interface = gr.Interface(
     fn=agent_interface,
     inputs=gr.Textbox(label="Enter a Topic", placeholder="e.g., AI developments, climate change..."),
@@ -126,4 +137,4 @@ interface = gr.Interface(
     """
 )
 
-interface.launch()
+interface.launch(share=True)