Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-"""Hugging Face Spaces entry point - handle backslash paths in src imports"""
+"""Hugging Face Spaces entry point - handle backslashes as literal filenames"""
 import os
 import sys
 import importlib.util
@@ -7,68 +7,61 @@ from pathlib import Path
 
 app_dir = Path("/app")
 
-# Add app_dir to sys.path FIRST
+# Add app_dir to sys.path
 sys.path.insert(0, str(app_dir))
 
-# Find and preload critical src modules
-src_dir = None
+print(f"App dir: {app_dir}")
+print(f"Contents:")
+
+# List all items - backslashes are literal in filenames!
+ui_app_path = None
 for item in app_dir.iterdir():
-    if item.is_dir() and "src" in item.name:
-        src_dir = item
-        break
+    print(f"  {item.name}")
+    # Look for the literal filename "ui\app.py"
+    if item.name == "ui\\app.py":
+        ui_app_path = item
 
-if src_dir:
-    print(f"Found src directory: {src_dir}")
-    sys.path.insert(0, str(src_dir))
+if ui_app_path and ui_app_path.exists():
+    print(f"\n Found ui\\app.py at: {ui_app_path}")
+    
+    # Read the file content and fix imports
+    with open(str(ui_app_path), 'r') as f:
+        code = f.read()
+    
+    # Replace problematic imports with direct module loading
+    code = code.replace(
+        "from src.models import",
+        "from models import"
+    ).replace(
+        "from src.",
+        "from "
+    )
     
-    # 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
+    # Preload src modules into sys.modules with correct names
+    src_files = [f for f in app_dir.iterdir() if str(f).startswith(str(app_dir) + "/src")]
+    for src_file in src_files:
+        if src_file.name.endswith(".py") and not src_file.name.startswith("_"):
+            module_name = src_file.stem
             try:
-                spec = importlib.util.spec_from_file_location(module_name, str(py_file))
+                spec = importlib.util.spec_from_file_location(module_name, str(src_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}")
-
-# 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]}")
     
+    # Execute the modified code
+    exec_globals = {"__name__": "__main__"}
     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
-            )
+        exec(compile(code, str(ui_app_path), "exec"), exec_globals)
     except Exception as e:
-        print(f"Error loading ui/app.py: {e}")
+        print(f"Error executing ui/app.py: {e}")
         import traceback
         traceback.print_exc()
 else:
-    print(f"ERROR: Could not find ui/app.py")
-    print(f"Contents of {app_dir}:")
-    for item in sorted(app_dir.iterdir()):
-        print(f"  {item}")
+    print(f"\n ERROR: Could not find ui\\app.py")
+    print(f"Looking for files with backslash in name...")
+    for item in app_dir.iterdir():
+        if "\\" in item.name:
+            print(f"  Found: {item.name}")