Update app.py

app.py

@@ -234,15 +234,28 @@ class SmartAgent:
         logging.info("SmartAgent resources cleaned up.")
 
 # --- Main Function ---
-def run_and_submit_all(profile: gr.OAuthProfile | None):
+def run_and_submit_all(request: gr.Request):
     """Run agent evaluation and submit results"""
-    if not profile:
-        logging.info("User not logged in.")
-        yield "Please Login to Hugging Face with the button.", pd.DataFrame([])
-        return
+    # Try to get username from request or fallback to space owner
+    username = None
+    
+    # Check for oauth profile
+    if hasattr(request, 'oauth_profile') and request.oauth_profile:
+        username = request.oauth_profile.username
+        logging.info(f"Found username via oauth_profile: {username}")
+    
+    # Fallback to space owner
+    if not username:
+        space_id = os.getenv("SPACE_ID", "")
+        if "/" in space_id:
+            username = space_id.split("/")[0]
+            logging.info(f"Using fallback username from SPACE_ID: {username}")
+        else:
+            logging.info("No authentication detected.")
+            yield "❌ Unable to detect logged-in user. Please make sure you're logged in to Hugging Face.", pd.DataFrame([])
+            return
 
-    username = f"{profile.username}"
-    logging.info(f"User logged in: {username}")
+    logging.info(f"Processing with username: {username}")
 
     api_url = DEFAULT_API_URL
     space_id = os.getenv("SPACE_ID") 
@@ -381,10 +394,9 @@ with gr.Blocks() as demo:
 
     run_button.click(
         fn=run_and_submit_all,
-        inputs=[gr.LoginButton()],
         outputs=[status_output, results_table],
         show_progress=True,
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)