initial commit

app.py

@@ -0,0 +1,76 @@
+import networkx as nx
+from smolagents import CodeAgent, HfApiModel, tool
+import gradio as gr
+
+# Define a tool for analyzing the Florentine Families graph
+@tool
+def analyze_florentine_graph(metric: str) -> str:
+    """
+    Analyzes the Florentine Families graph based on the chosen metric.
+
+    Args:
+        metric (str): The centrality or analysis metric to calculate (e.g., 'degree', 'betweenness', 'closeness').
+
+    Returns:
+        str: A textual analysis of the chosen metric for the graph.
+    """
+    # Load the Florentine Families graph
+    graph = nx.florentine_families_graph()
+
+    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'."
+
+    # Format the results
+    analysis = f"Analysis of {metric} centrality:\n"
+    for node, value in centrality.items():
+        analysis += f"- {node}: {value:.3f}\n"
+
+    return analysis
+
+# Create the CodeAgent
+class FlorentineGraphAgentUI:
+    def __init__(self):
+        self.model = HfApiModel("meta-llama/Llama-3.3-70B-Instruct")
+        self.agent = CodeAgent(
+            tools=[analyze_florentine_graph],
+            model=self.model,
+            additional_authorized_imports=["networkx"]
+        )
+
+    def process_query(self, query: str) -> str:
+        """Process a natural language query to analyze the graph."""
+        return self.agent.run(query)
+
+    def launch(self):
+        """Launch the Gradio interface."""
+        interface = gr.Interface(
+            fn=self.process_query,
+            inputs=gr.Textbox(
+                label="What would you like to analyze about the Florentine Families graph?",
+                placeholder="e.g., 'Analyze degree centrality'"
+            ),
+            outputs=gr.Textbox(label="Graph Analysis Results"),
+            title="Florentine Families Graph Analysis",
+            description="""
+            This app analyzes the Florentine Families graph using various centrality metrics. 
+            You can request analyses like degree centrality, betweenness centrality, or closeness centrality.
+            """,
+            examples=[
+                ["Analyze degree centrality"],
+                ["Analyze betweenness centrality"],
+                ["Analyze closeness centrality"]
+            ],
+            cache_examples=False
+        )
+        interface.launch()
+
+if __name__ == "__main__":
+    # Initialize and launch the app
+    app = FlorentineGraphAgentUI()
+    app.launch()