Update app.py

app.py

@@ -5,24 +5,28 @@ import inspect
 import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
 # (Keep Constants as is)
+from dotenv import load_dotenv
+load_dotenv()
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+if os.getenv('HF_TOKEN'):
+    os.environ['HF_TOKEN'] = os.getenv('HF_TOKEN')
+else:
+    print("HF_TOKEN environment variable not set!")
 
-
-# --- Basic Agent Definition ---
+# --- Basic Agent Definition ---S
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
-    def __init__(self, GOOGLE_API_KEY):
-        self.GOOGLE_API_KEY = GOOGLE_API_KEY
+    def __init__(self, api_key):
+        self.api_key = api_key
         print("BasicAgent initialized.")
         # Initialize the model
         #model = HfApiModel()
         model = OpenAIServerModel(
-        model_id="gemini-2.5-flash",
-    # Google Gemini OpenAI-compatible API base URL
-        api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
-        api_key=GOOGLE_API_KEY
-)
+            model_id="gemma2:latest",
+            api_base="http://localhost:11434/v1/",
+            api_key=api_key if api_key else "dummy-key-for-ollama"
+        )
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
         # Initialize Agent
@@ -36,7 +40,7 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
-def run_and_submit_all(profile: gr.OAuthProfile | None, openai_key: str):
+def run_and_submit_all(openai_key: str):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -44,12 +48,21 @@ def run_and_submit_all(profile: gr.OAuthProfile | None, openai_key: str):
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
 
-    if profile:
-        username= f"{profile.username}"
-        print(f"User logged in: {username}")
-    else:
-        print("User not logged in.")
-        return "Please Login to Hugging Face with the button.", None
+    # Get user info from Gradio's login state
+    try:
+        import gradio as gr
+        user_info = gr.user_info()
+        if user_info and hasattr(user_info, 'username'):
+            username = user_info.username
+            print(f"User logged in: {username}")
+        else:
+            print("User not logged in.")
+            return "Please Login to Hugging Face with the button.", None
+    except Exception as e:
+        print(f"Error getting user info: {e}")
+        # Fallback: try to get username from environment or use a default
+        username = os.getenv("USER", "anonymous")
+        print(f"Using fallback username: {username}")
 
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"