Update app.py

app.py

@@ -34,14 +34,46 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-final_answer = FinalAnswerTool()
-model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-custom_role_conversions=None,
+# Load AI Regulation Model (No HF Token required)
+ai_regulation_model = HfApiModel(
+    max_tokens=500,
+    temperature=0.5,
+    model_id="Orcawise/eu_ai_act_using_finetuned_gemma"
 )
 
+@tool
+def ai_act_search_tool(query: str, max_results: int = 5) -> dict:
+    """
+    Searches for AI Act insights using an AI model and DuckDuckGo.
+
+    Arguments:
+        query: (str) The search query related to the AI Act.
+        max_results: (int) The maximum number of search results to return.
+
+    Returns:
+        (dict) A dictionary containing:
+            - "ai_insight": AI-generated insights on the AI Act query.
+            - "articles": A list of URLs to relevant articles.
+    """
+    # Step 1: Get AI Model Insights
+    ai_response = ai_regulation_model.generate(prompt=f"Provide insights on: {query} related to the AI Act.")
+    ai_insight = ai_response["generated_text"] if isinstance(ai_response, dict) and "generated_text" in ai_response else "No AI insights available."
+
+    # Step 2: Perform DuckDuckGo Search
+    try:
+        with DDGS() as ddgs:
+            search_results = ddgs.text(query + " AI Act", max_results)
+        articles = [{"title": result["title"], "link": result["href"]} for result in search_results] if search_results else []
+    except Exception as e:
+        articles = []
+        print(f"Error in DuckDuckGo search: {e}")
+
+    # Step 3: Return Combined Results
+    return {
+        "ai_insight": ai_insight,
+        "articles": articles
+    }
+
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)