Fix Gradio 4.44.x schema bug by removing progress and disabling API

app.py

@@ -14,7 +14,7 @@ sys.path.insert(0, str(Path(__file__).parent))
 sys.path.insert(0, str(Path(__file__).parent / "teacher_agent_dev"))
 sys.path.insert(0, str(Path(__file__).parent / "student_agent_dev"))
 
-def run_comparison(iterations: int, seed: int, use_deterministic: bool, device: str, progress=None):
+def run_comparison(iterations: int, seed: int, use_deterministic: bool, device: str):
     """
     Run strategy comparison with LM Student.
     
@@ -23,7 +23,6 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
         seed: Random seed (ignored if deterministic)
         use_deterministic: Use fixed seed=42
         device: 'cpu' or 'cuda' (GPU)
-        progress: Gradio progress tracker
     """
     
     # Set device environment variable for subprocess
@@ -33,18 +32,8 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
             import torch
             if not torch.cuda.is_available():
                 device = "cpu"
-                if progress is not None:
-                    try:
-                        progress(0.0, desc="⚠️ GPU not available, using CPU...")
-                    except (AttributeError, IndexError, TypeError):
-                        pass
         except ImportError:
             device = "cpu"
-            if progress is not None:
-                try:
-                    progress(0.0, desc="⚠️ PyTorch not available, using CPU...")
-                except (AttributeError, IndexError, TypeError):
-                    pass
         except Exception:
             device = "cpu"
     
@@ -64,13 +53,6 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
         cmd.extend(["--seed", str(int(seed))])
     
     try:
-        # Safe progress update - wrap in try/except for Gradio compatibility
-        if progress is not None:
-            try:
-                progress(0.1, desc="Starting comparison...")
-            except (AttributeError, IndexError, TypeError):
-                pass
-        
         # Ensure environment variables are passed to subprocess
         env = os.environ.copy()
         env["CUDA_DEVICE"] = os.environ.get("CUDA_DEVICE", device)
@@ -90,13 +72,6 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
         # Combine outputs
         full_output = f"=== STDOUT ===\n{stdout_text}\n\n=== STDERR ===\n{stderr_text}"
         
-        # Safe progress update
-        if progress is not None:
-            try:
-                progress(0.9, desc="Processing results...")
-            except (AttributeError, IndexError, TypeError):
-                pass
-        
         if result.returncode != 0:
             return f"❌ Error occurred:\n{full_output}", None
         
@@ -114,11 +89,6 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
                 break
         
         if plot_path:
-            if progress is not None:
-                try:
-                    progress(1.0, desc="Complete!")
-                except (AttributeError, IndexError, TypeError):
-                    pass
             return f"✅ Comparison complete!\n\n{stdout_text}", str(plot_path)
         else:
             # Return output even if plot not found (might still be useful)
@@ -243,7 +213,7 @@ with gr.Blocks(title="MentorFlow - Strategy Comparison") as demo:
     """)
 
 if __name__ == "__main__":
-    # For Hugging Face Spaces, Gradio auto-detects the environment
-    # No need to set share or server_name/port explicitly
-    demo.launch()
+    # Disable API generation to avoid Gradio 4.44.x schema bug
+    # This bypasses the TypeError: argument of type 'bool' is not iterable
+    demo.launch(show_api=False)