Fix Gradio 4.44.0 bugs - upgrade to 4.44.1 and fix launch

README.md

@@ -4,7 +4,7 @@ emoji: 🎓
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 4.44.0
+sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 license: mit

README_HF_SPACE.md

@@ -4,7 +4,7 @@ emoji: 🎓
 colorFrom: blue
 colorTo: purple
 sdk: gradio
-sdk_version: 4.44.0
+sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 license: mit

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=gr.Progress(track_tqdm=True)):
+def run_comparison(iterations: int, seed: int, use_deterministic: bool, device: str, progress=None):
     """
     Run strategy comparison with LM Student.
     
@@ -33,16 +33,18 @@ 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="⚠️ GPU not available, using CPU...")
+                    progress(0.0, desc="⚠️ PyTorch not available, using CPU...")
                 except (AttributeError, IndexError, TypeError):
                     pass
-        except ImportError:
-            device = "cpu"
-            try:
-                progress(0.0, desc="⚠️ PyTorch not available, using CPU...")
-            except (AttributeError, IndexError, TypeError):
-                pass
         except Exception:
             device = "cpu"
     
@@ -63,10 +65,11 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
     
     try:
         # Safe progress update - wrap in try/except for Gradio compatibility
-        try:
-            progress(0.1, desc="Starting comparison...")
-        except (AttributeError, IndexError, TypeError):
-            pass
+        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()
@@ -88,10 +91,11 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
         full_output = f"=== STDOUT ===\n{stdout_text}\n\n=== STDERR ===\n{stderr_text}"
         
         # Safe progress update
-        try:
-            progress(0.9, desc="Processing results...")
-        except (AttributeError, IndexError, TypeError):
-            pass
+        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
@@ -110,10 +114,11 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
                 break
         
         if plot_path:
-            try:
-                progress(1.0, desc="Complete!")
-            except (AttributeError, IndexError, TypeError):
-                pass
+            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)
@@ -238,5 +243,7 @@ with gr.Blocks(title="MentorFlow - Strategy Comparison") as demo:
     """)
 
 if __name__ == "__main__":
-    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)
+    # For Hugging Face Spaces, Gradio auto-detects the environment
+    # No need to set share or server_name/port explicitly
+    demo.launch()
 

requirements.txt

@@ -15,8 +15,8 @@ seaborn>=0.12.0
 # Progress bars
 tqdm>=4.65.0
 
-# Gradio for web interface
-gradio>=4.44.0
+# Gradio for web interface (updated for security fixes)
+gradio>=4.44.1
 
 # Additional utilities
 scipy>=1.10.0

requirements_hf.txt

@@ -16,7 +16,7 @@ seaborn>=0.12.0
 tqdm>=4.65.0
 
 # Gradio for web interface (updated for security fixes)
-gradio>=4.44.0
+gradio>=4.44.1
 
 # Additional utilities
 scipy>=1.10.0