Update app.py

app.py

@@ -1,3 +1,5 @@
+# Imports
+
 import os
 import json
 import smolagents
@@ -10,8 +12,8 @@ import difflib
 import openai
 from langchain_community.utilities.wikipedia import WikipediaAPIWrapper
 import gradio as gr
-print("smolagents version:", smolagents.__version__)
-print("openai version:", openai.__version__)
+
+# Settings
 
 REPO_ID_TECHSPARK_STAFF = "aslan-ng/CMU_TechSpark_Staff"
 REPO_ID_TECHSPARK_COURSES = "aslan-ng/CMU_TechSpark_Courses"
@@ -24,6 +26,8 @@ login(hf_token)
 
 NUMERIC_PROFILE = ["Laser Cutting",	"Wood Working",	"Wood CNC",	"Metal Machining",	"Metal CNC",	"3D Printer",	"Welding",	"Electronics"]
 
+# Load Data
+
 def load_data_from_huggingface():
     """
     Loads data from HuggingFace.
@@ -52,6 +56,8 @@ def load_data_from_huggingface():
 
 staff_df, courses_df, tools_df, nodes_df, edges_df = load_data_from_huggingface()
 
+# General Functions
+
 def vector_1st_distance(x: list, y: list):
     """
     Calculate the average 1st distance between two vectors.
@@ -102,6 +108,8 @@ def skill_score(
         y.append(electronics)
     return vector_1st_distance(x, y)
 
+# Staff Functions
+    
 def all_staff():
     """
     Return a list of all staff.
@@ -246,6 +254,8 @@ class FindSuitableStaff(smolagents.tools.Tool):
         staff_profiles = [get_staff_profile(name) for name in names]
         return json.dumps(staff_profiles)
 
+# Course Functions
+
 def all_courses_code():
     """
     Return a list of all course codes.
@@ -406,6 +416,8 @@ class FindSuitableCourses(smolagents.tools.Tool):
         options = [get_course_profile(course) for course in matches]
         return json.dumps(options)
 
+# Machines and Tools Functions
+
 def all_tools():
     """
     Return a list of all tools and machines.
@@ -488,6 +500,8 @@ class CheckMachineAccessibility(smolagents.tools.Tool):
     def forward(self, name: str) -> str:
         return json.dumps(is_tool_accessible(name))
 
+# Wikipedia Functions
+
 class WikipediaSearch(smolagents.Tool):
     """
     Create tool for searching Wikipedia
@@ -506,6 +520,8 @@ class WikipediaSearch(smolagents.Tool):
         answer = wikipedia_api.run(query)
         return answer
 
+# Map Functions
+
 def all_nodes():
     """
     Return a list of all nodes name.
@@ -553,36 +569,6 @@ def load_graph(nodes_df, edges_df):
 
 G = load_graph(nodes_df, edges_df)
 
-def show_graph():
-    import matplotlib.pyplot as plt
-
-    pos = {
-      row["ID"]: (row["X"], row["Y"])
-      for _, row in nodes_df.iterrows()
-    }
-
-    labels = {
-        row["ID"]: row["Name"]
-        for _, row in nodes_df.iterrows()
-    }
-
-    plt.figure(figsize=(8, 6))
-
-    nx.draw(
-        G,
-        pos=pos,
-        #labels=labels,  # ID or Name?
-        with_labels=True,
-        node_size=800,
-        node_color="lightblue",
-        font_size=10,
-        font_weight="bold",
-        edge_color="gray"
-    )
-
-    plt.axis("equal")
-    plt.show()
-
 def path_finder(destination: int, source: int):
     try:
         path = nx.shortest_path(G, source=source, target=destination)
@@ -694,6 +680,8 @@ class PathFinding(smolagents.tools.Tool):
     def forward(self, destination: str) -> str:
         return path_human(destination, source=None)
 
+# Agent
+
 techspark_definition = """
 TechSpark is the largest makerspace at CMU (Carnegie Mellon University), located in the College of Engineering.  
 Its mission is to promote a vibrant, student-centric making culture to enhance educational, extracurricular, and research activities across the entire campus community.
@@ -740,13 +728,15 @@ agent = smolagents.CodeAgent(
     verbosity_level=2,  # show steps in logs for class demo
 )
 
+# UI
+
 with gr.Blocks() as demo:
 
     # Centered title and description using HTML
     gr.HTML("""
         <div style="text-align: center; font-family: 'Arial', sans-serif;">
             <h1 style="color:#1f77b4; margin-bottom: 20px; font-weight: 300;">
-                🤖 TechSpark AI Assistant
+                🔧 TechSpark AI Assistant 🤖
             </h1>
 
             <p style="margin-top: 0; font-weight: 300; font-size: 16px; color:#555;">