Update app.py

app.py

@@ -1,4 +1,3 @@
-"""Basic Agent Evaluation Runner - Cleaned Version"""
 import os
 import gradio as gr
 import requests
@@ -10,9 +9,8 @@ from tools import build_graph
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
-    """A langgraph agent using CSV-based vector store."""
     
-    def __init__(self, csv_file_path: str = "embeddings.csv"):
+    def __init__(self, csv_file_path: str = "embedding_database.csv"):
         print("BasicAgent initialized.")
         self.graph = build_graph(provider="groq", csv_file_path=csv_file_path)
     
@@ -40,7 +38,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
     """
-    # Check authentication
     if not profile:
         print("User not logged in.")
         return "Please Login to Hugging Face with the button.", None
@@ -48,16 +45,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     username = f"{profile.username}"
     print(f"User logged in: {username}")
     
-    # Determine space info
     space_id = os.getenv("SPACE_ID")
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else "Local Development"
     
-    # API URLs
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
     
-    # Initialize Agent
     try:
         agent = BasicAgent()  # Make sure your CSV file is named "embeddings.csv" or update the path
     except Exception as e:
@@ -79,7 +73,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print(f"Error fetching questions: {e}")
         return f"Error fetching questions: {e}", None
     
-    # Run Agent on Questions
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
@@ -113,7 +106,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not answers_payload:
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
     
-    # Submit Results
     submission_data = {
         "username": username.strip(), 
         "agent_code": agent_code, 
@@ -144,7 +136,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         results_df = pd.DataFrame(results_log)
         return status_message, results_df
 
-# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""
@@ -170,7 +161,6 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)
     
-    # Environment info
     space_host = os.getenv("SPACE_HOST")
     space_id = os.getenv("SPACE_ID")