Update app.py

app.py

@@ -1,21 +1,24 @@
+# Validate the updated code by checking for syntax and logical errors
+
+# Importing necessary modules for testing
 import networkx as nx
 from smolagents import CodeAgent, HfApiModel, tool
 import gradio as gr
 
-# Define a tool for analyzing the Florentine Families graph
+# Test tool definition
 @tool
 def analyze_florentine_graph(metric: str) -> str:
     """
     Analyze the Florentine Families graph using a specified centrality metric.
 
     Args:
-        metric (str): The name of the centrality metric to calculate. Supported values are:
-            - 'degree': Calculate degree centrality for each node.
-            - 'betweenness': Calculate betweenness centrality for each node.
-            - 'closeness': Calculate closeness centrality for each node.
+        metric (str): The centrality metric to calculate. Supported values are:
+            - 'degree': Degree centrality
+            - 'betweenness': Betweenness centrality
+            - 'closeness': Closeness centrality
 
     Returns:
-        str: A formatted string displaying the centrality values for each node in the graph.
+        str: A formatted string showing the centrality values for each node in the graph.
     """
     # Load the Florentine Families graph
     graph = nx.florentine_families_graph()
@@ -36,7 +39,7 @@ def analyze_florentine_graph(metric: str) -> str:
 
     return analysis
 
-# Create the CodeAgent
+# Test Agent Class
 class FlorentineGraphAgentUI:
     def __init__(self):
         self.model = HfApiModel("meta-llama/Llama-3.3-70B-Instruct")
@@ -73,7 +76,11 @@ class FlorentineGraphAgentUI:
         )
         interface.launch()
 
-if __name__ == "__main__":
-    # Initialize and launch the app
-    app = FlorentineGraphAgentUI()
-    app.launch()
+# Instance of the app to simulate the behavior
+app = FlorentineGraphAgentUI()
+
+# Simulate a query
+try:
+    print(app.process_query("Analyze degree centrality"))
+except Exception as e:
+    print(f"Error during execution: {e}")