deepseek point-of-view

app.py

@@ -1,64 +1,42 @@
 import networkx as nx
 from smolagents import CodeAgent, HfApiModel, tool, GradioUI
-from typing import Literal
 
+# Define a tool for analyzing the Florentine Families graph
 @tool
-def analyze_florentine_graph(metric: Literal["degree", "betweenness", "closeness"]) -> str:
+def analyze_florentine_graph(metric: str) -> str:
     """
-    Analyzes the Florentine Families social network graph using centrality metrics.
-    
+    Analyzes the Florentine Families graph based on the chosen centrality metric.
+
     Args:
-        metric: The centrality metric to calculate. Must be one of:
-               - "degree": Measures direct connections
-               - "betweenness": Measures bridge/broker positions
-               - "closeness": Measures average distance to all other nodes
+        metric: The centrality metric to calculate. Valid options: 'degree', 'betweenness', or 'closeness'.
+
+    Returns:
+        A textual analysis of the chosen metric for the graph with formatted numerical results.
     """
     graph = nx.florentine_families_graph()
-    
-    centrality_funcs = {
-        "degree": nx.degree_centrality,
-        "betweenness": nx.betweenness_centrality,
-        "closeness": nx.closeness_centrality
-    }
-    
-    if metric not in centrality_funcs:
-        return f"Invalid metric. Please choose from: {', '.join(centrality_funcs.keys())}"
-        
-    centrality = centrality_funcs[metric](graph)
-    
-    # Sort by centrality value for clearer output
-    sorted_values = sorted(centrality.items(), key=lambda x: x[1], reverse=True)
-    
-    analysis = f"\nTop families by {metric} centrality:\n"
-    for family, value in sorted_values:
-        analysis += f"- {family}: {value:.3f}\n"
-        
+
+    if metric == "degree":
+        centrality = nx.degree_centrality(graph)
+    elif metric == "betweenness":
+        centrality = nx.betweenness_centrality(graph)
+    elif metric == "closeness":
+        centrality = nx.closeness_centrality(graph)
+    else:
+        return "Invalid metric. Please choose 'degree', 'betweenness', or 'closeness'."
+
+    analysis = f"Analysis of {metric} centrality:\n"
+    for node, value in centrality.items():
+        analysis += f"- {node}: {value:.3f}\n"
     return analysis
 
-# Initialize agent with model and tool
+# Initialize the agent with proper configuration
 model = HfApiModel("meta-llama/Llama-3.3-70B-Instruct")
 agent = CodeAgent(
     tools=[analyze_florentine_graph],
     model=model,
     additional_authorized_imports=["networkx"],
+    add_base_tools=True
 )
 
-# Launch Gradio interface using smolagents' built-in UI
-if __name__ == "__main__":
-    ui = GradioUI(
-        agent=agent,
-        title="Florentine Families Network Analysis",
-        description="""
-        Analyze the historical Florentine families social network using various centrality metrics.
-        Try asking questions like:
-        - "Who were the most central families by degree centrality?"
-        - "Which families acted as brokers (high betweenness)?"
-        - "Analyze the network using closeness centrality"
-        """,
-        examples=[
-            "Analyze the network using degree centrality",
-            "Who were the main broker families according to betweenness centrality?",
-            "Show me the closeness centrality analysis"
-        ]
-    )
-    ui.launch()
+# Launch the Gradio interface using smolagents' built-in UI component
+GradioUI(agent).launch()