Update app.py

app.py

@@ -97,32 +97,62 @@ ANSWERS = [
 def get_username():
     """Get the username of the logged-in user"""
     try:
-        return os.environ.get("SPACE_AUTHOR")
+        # Try multiple methods to get the username
+        username = os.environ.get("SPACE_AUTHOR")
+        
+        # If that doesn't work, try using the HfApi
+        if not username:
+            try:
+                from huggingface_hub import HfApi
+                api = HfApi()
+                user_info = api.whoami()
+                username = user_info.get("name", None)
+            except:
+                pass
+                
+        return username
     except:
         return None
 
 def check_login_status():
     """Check if the user is logged in"""
-    username = get_username()
-    if username:
-        return f"Logged in as {username}"
-    else:
-        return "Not logged in. Please log in to submit."
+    try:
+        # Try using HfApi directly
+        from huggingface_hub import HfApi
+        api = HfApi()
+        user_info = api.whoami()
+        username = user_info.get("name", None)
+        
+        if username:
+            return f"Logged in as {username}"
+        else:
+            return "Not logged in. Please log in to submit."
+    except Exception as e:
+        return f"Error checking login status: {str(e)}"
 
 def run_and_submit_all():
     """Run the agent on all questions and submit the answers"""
-    username = get_username()
-    if not username:
-        return "Please log in to submit your answers.", None
-    
-    # Get the Space name
-    space_name = os.environ.get("SPACE_ID", "")
-    
-    # Create the code URL
-    code_url = f"https://huggingface.co/spaces/{space_name}/tree/main"
-    
-    # Submit to the API
     try:
+        # Get username directly from HfApi
+        from huggingface_hub import HfApi
+        api = HfApi()
+        user_info = api.whoami()
+        username = user_info.get("name", None)
+        
+        if not username:
+            return "Please log in to submit your answers.", None
+        
+        # Get the Space name
+        space_name = os.environ.get("SPACE_ID", "")
+        if not space_name:
+            # Try to get it from the URL if available
+            space_name = f"{username}/{os.environ.get('SPACE_TITLE', 'gaia-submission')}"
+        
+    
+        # Create the code URL
+        code_url = f"https://huggingface.co/spaces/{space_name}/tree/main"
+        
+        # Submit to the API
         api_url = "https://agents-course-unit4-scoring.hf.space/submit"
         payload = {
             "username": username,
@@ -157,6 +187,7 @@ def run_and_submit_all():
             return submission_result, df
         else:
             return f"Error: {response.status_code} - {response.text}", None
+            
     except Exception as e:
         return f"Error: {str(e)}", None