Update app.py

app.py

@@ -13,27 +13,49 @@ from Gradio_UI import GradioUI
 @tool
 def search_latest_ai_models(topic: str) -> str:
     """
-    Uses DuckDuckGo to search for the latest AI/GenAI advancements and returns it in tabular format.
-    
-    Args:
-        topic: The topic to search, e.g., "latest AI models June 2025"
+    Searches for latest AI/GenAI model announcements using DuckDuckGo and returns results in tabular format.
 
+    Args:
+        topic: Topic to search (e.g., 'latest Generative AI models 2025')
     Returns:
-        A markdown table of recent AI model announcements, their creators, purposes, and benefits.
+        Markdown table summarizing AI model, company, and purpose.
     """
     search_tool = DuckDuckGoSearchTool()
     results = search_tool(topic)
 
-    # Example post-processing (this can be replaced by better parsing in real-time)
-    # Dummy entries for now (you can customize parsing later)
-    table_data = [
-        {"Model": "Qwen 2.5", "Company": "Alibaba", "Purpose": "Code generation", "Benefit": "Improved coding accuracy"},
-        {"Model": "Gemini 1.5", "Company": "Google DeepMind", "Purpose": "Multi-modal reasoning", "Benefit": "Handles longer context, integrates video"},
-        {"Model": "Veo 2", "Company": "Google", "Purpose": "Text-to-video", "Benefit": "High-fidelity, longer videos"},
-    ]
-    
-    df = pd.DataFrame(table_data)
-    return df.to_markdown(index=False)
+    # Process first 5 results only for simplicity
+    table_data = []
+    for item in results[:5]:
+        # Try to split headline to get a rough model and org
+        title = item.get("title", "")
+        link = item.get("link", "")
+        snippet = item.get("body", "")
+
+        model = title.split(" ")[0]
+        company = "Unknown"
+        if "Google" in snippet or "Google" in title:
+            company = "Google"
+        elif "OpenAI" in snippet or "OpenAI" in title:
+            company = "OpenAI"
+        elif "Anthropic" in snippet or "Anthropic" in title:
+            company = "Anthropic"
+        elif "Alibaba" in snippet or "Alibaba" in title:
+            company = "Alibaba"
+
+        table_data.append({
+            "Model": model,
+            "Company": company,
+            "Purpose": snippet[:100] + "..." if snippet else "N/A",
+            "Link": link
+        })
+
+    # Convert to markdown table
+    markdown = "| Model | Company | Purpose | Link |\n|-------|---------|---------|------|\n"
+    for row in table_data:
+        markdown += f"| {row['Model']} | {row['Company']} | {row['Purpose']} | [Link]({row['Link']}) |\n"
+
+    return markdown
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: