Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-"""Hugging Face Spaces entry point - handles backslash filenames"""
+"""Hugging Face Spaces entry point - handle backslash paths in src imports"""
 import os
 import sys
 import importlib.util
@@ -7,53 +7,68 @@ from pathlib import Path
 
 app_dir = Path("/app")
 
-# Find ui\app.py (with backslash in the filename on Linux)
-ui_app_files = list(app_dir.glob("**/app.py"))
-ui_app_path = None
+# Add app_dir to sys.path FIRST
+sys.path.insert(0, str(app_dir))
 
-for f in ui_app_files:
-    # Look for ui/app.py or ui\app.py
-    if "ui" in str(f) and f.name == "app.py":
-        # Skip /app/app.py (the entry point itself)
-        if f.parent.name == "ui" or "ui" in f.parent.name:
-            ui_app_path = f
-            break
+# Find and preload critical src modules
+src_dir = None
+for item in app_dir.iterdir():
+    if item.is_dir() and "src" in item.name:
+        src_dir = item
+        break
+
+if src_dir:
+    print(f"Found src directory: {src_dir}")
+    sys.path.insert(0, str(src_dir))
+    
+    # Preload critical modules
+    for py_file in src_dir.glob("*.py"):
+        if py_file.name != "__init__.py" and not py_file.name.startswith("_"):
+            module_name = py_file.stem
+            try:
+                spec = importlib.util.spec_from_file_location(module_name, str(py_file))
+                module = importlib.util.module_from_spec(spec)
+                sys.modules[module_name] = module
+                spec.loader.exec_module(module)
+                print(f"  Loaded {module_name}")
+            except Exception as e:
+                print(f"  Warning loading {module_name}: {e}")
 
-if not ui_app_path:
-    # Last resort - manually check common locations
-    possible_paths = [
-        app_dir / "ui" / "app.py",
-        app_dir / "ui\\app.py",  # With backslash
-    ]
-    for p in possible_paths:
-        if p.exists():
-            ui_app_path = p
+# Find ui/app.py
+ui_app_path = None
+for item in app_dir.iterdir():
+    if item.is_dir() and "ui" in item.name:
+        ui_app = item / "app.py"
+        if ui_app.exists():
+            ui_app_path = ui_app
             break
 
 if ui_app_path and ui_app_path.exists():
     print(f"Found ui/app.py at: {ui_app_path}")
+    print(f"sys.path: {sys.path[:3]}")
     
-    # Add app_dir to sys.path so imports work
-    sys.path.insert(0, str(app_dir))
-    
-    # Load the module
-    spec = importlib.util.spec_from_file_location("ui_app", str(ui_app_path))
-    ui_app_module = importlib.util.module_from_spec(spec)
-    sys.modules["ui_app"] = ui_app_module
-    
-    # Execute the module
-    spec.loader.exec_module(ui_app_module)
-    
-    if __name__ == "__main__":
-        demo = ui_app_module.create_gradio_app()
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=int(os.environ.get("PORT", 7860)),
-            show_error=True
-        )
+    try:
+        # Load the module
+        spec = importlib.util.spec_from_file_location("ui_app", str(ui_app_path))
+        ui_app_module = importlib.util.module_from_spec(spec)
+        sys.modules["ui_app"] = ui_app_module
+        
+        # Execute the module - now imports should work
+        spec.loader.exec_module(ui_app_module)
+        
+        if __name__ == "__main__":
+            demo = ui_app_module.create_gradio_app()
+            demo.launch(
+                server_name="0.0.0.0",
+                server_port=int(os.environ.get("PORT", 7860)),
+                show_error=True
+            )
+    except Exception as e:
+        print(f"Error loading ui/app.py: {e}")
+        import traceback
+        traceback.print_exc()
 else:
     print(f"ERROR: Could not find ui/app.py")
-    print(f"App directory: {app_dir}")
-    print(f"Files found:")
-    for f in sorted(app_dir.glob("**/*.py")):
-        print(f"  {f}")
+    print(f"Contents of {app_dir}:")
+    for item in sorted(app_dir.iterdir()):
+        print(f"  {item}")