Update app.py

app.py

@@ -8,24 +8,6 @@ import matplotlib.pyplot as plt
 import numpy as np
 from Gradio_UI import GradioUI
 
-@tool
-def plot_omocentric_circles(distance: float, n: int)-> None:
-    """A tool that plots n omocentric circles with a given seed distance, each in a different color.
-    Args:
-        distance: Distance between consecutive circles.
-        n: Number of circles to plot.
-    """
-    fig, ax = plt.subplots()
-    ax.set_aspect('equal')
-    
-    colors = plt.cm.viridis(np.linspace(0, 1, n))  # Generate n distinct colors
-
-    for i in range(n):
-        ax.add_patch(plt.Circle((0, 0), (i + 1) * seed, fill=False, color=colors[i]))
-
-    plt.xlim(-n * seed, n * seed)
-    plt.ylim(-n * seed, n * seed)
-    plt.show()
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -54,12 +36,57 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+#########################################################################
+# NEW TOOL: GET CULTURAL INFO (via Wikipedia)
+#########################################################################
+@tool
+def get_cultural_info(topic: str) -> str:
+    """
+    A tool that retrieves cultural or general info from Wikipedia for a given topic.
+    Args:
+        topic: A string representing the subject to lookup (e.g., 'Renaissance art').
+    Returns:
+        A short summary text from Wikipedia for the specified topic.
+    """
+    try:
+        # Query the Wikipedia API for an extract
+        # Using the "summary" approach with 'exintro' to get the first paragraph
+        url = (
+            "https://en.wikipedia.org/w/api.php?"
+            "action=query&prop=extracts&exintro&explaintext&format=json&titles=" + topic
+        )
+        response = requests.get(url)
+        data = response.json()
+        
+        # Extract the page data
+        pages = data.get("query", {}).get("pages", {})
+        if not pages:
+            return f"No Wikipedia pages found for topic: {topic}"
+
+        # Wikipedia returns a dictionary keyed by page ID
+        page_id = next(iter(pages))
+        page_content = pages[page_id]
+
+        # Check for missing page or missing extract
+        if "missing" in page_content:
+            return f"No page found on Wikipedia for topic: {topic}"
+
+        extract = page_content.get("extract", "")
+        if not extract:
+            return f"No extract available for '{topic}'."
+
+        return extract.strip()
+
+    except Exception as e:
+        return f"Error retrieving cultural info for '{topic}': {str(e)}"
+#########################################################################
+
 
 final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',# it is possible that this model may be overloaded
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
@@ -72,7 +99,14 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool(), image_generation_tool, plot_omocentric_circles], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        DuckDuckGoSearchTool(), 
+        VisitWebpageTool(), 
+        get_current_time_in_timezone,
+        get_cultural_info,          # <-- Our new cultural info tool
+        image_generation_tool
+    ],    
     max_steps=6,
     verbosity_level=1,
     grammar=None,