feat: implement dynamic environment detection to handle local and Hugging Face deployment logic

app.py

@@ -15,10 +15,16 @@ from langgraph.graph.message import add_messages
 from langgraph.prebuilt import tools_condition, ToolNode
 from langchain_core.messages import AnyMessage, HumanMessage, SystemMessage, ToolMessage
 from langchain_groq import ChatGroq
-# --- LOCAL RUN SETTINGS (Using langchain_ollama for local Ollama support) ---
-from langchain_ollama import ChatOllama
-# --- HUGGINGFACE SETTINGS (Uncomment for HF if needed, but langchain_ollama is preferred) ---
-# from langchain_community.chat_models import ChatOllama
+# Detect if running locally or on Hugging Face Spaces
+IS_LOCAL = os.getenv("SPACE_ID") is None
+
+if IS_LOCAL:
+    from langchain_ollama import ChatOllama
+else:
+    try:
+        from langchain_ollama import ChatOllama
+    except ImportError:
+        from langchain_community.chat_models import ChatOllama
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -144,16 +150,16 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
     """
-    # --- LOCAL RUN SETTINGS ---
-    # Set a default username for local testing
-    username = "local_user" 
-
-    # --- HUGGINGFACE SETTINGS (Comment out above and uncomment below for HF) ---
-    # if profile:
-    #     username = f"{profile.username}"
-    #     print(f"{username} logged in")
-    # else:
-    #     return "Please login to Hugging Face", None
+    if IS_LOCAL:
+        username = "local_user"
+        print(f"Running in LOCAL MODE. Using default username: {username}")
+    else:
+        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
 
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
@@ -165,11 +171,13 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
-    # --- LOCAL RUN SETTINGS ---
-    space_id = os.getenv("SPACE_ID", "local-test-space")
-    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id != "local-test-space" else "local-execution"
-    # --- END LOCAL RUN SETTINGS ---
-
+    if IS_LOCAL:
+        space_id = "local-test-space"
+        agent_code = "local-execution"
+    else:
+        space_id = os.getenv("SPACE_ID")
+        agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
+    
     print(f"Agent code link: {agent_code}")
 
     # 2. Fetch Questions
@@ -207,12 +215,11 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
 
-        # --- LOCAL RUN SETTINGS: Skip questions with files ---
-        if file_name:
+        # Skip questions with files if running locally
+        if IS_LOCAL and file_name:
             print(f"Skipping task {task_id} because it requires a file: {file_name}")
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": "SKIPPED (Local run - No files)"})
             continue
-        # --- END LOCAL RUN SETTINGS ---
 
         try:
             submitted_answer = agent(question_text, file_name)
@@ -322,9 +329,8 @@ with gr.Blocks() as demo:
                 """
             )
 
-            # --- LOCAL RUN SETTINGS (Comment out for HF) ---
-            # gr.LoginButton()
-            # --- END LOCAL RUN SETTINGS ---
+            if not IS_LOCAL:
+                gr.LoginButton()
 
             run_button = gr.Button("Run Full Evaluation & Submit All Answers", variant="primary")