Upload app.py with huggingface_hub

app.py

@@ -1,67 +1,81 @@
 #!/usr/bin/env python3
-"""Hugging Face Spaces entry point - handle backslashes as literal filenames"""
+"""Hugging Face Spaces - fix backslash filenames"""
 import os
 import sys
 import importlib.util
 from pathlib import Path
 
 app_dir = Path("/app")
-
-# Add app_dir to sys.path
 sys.path.insert(0, str(app_dir))
 
-print(f"App dir: {app_dir}")
-print(f"Contents:")
+print(f"Starting app from {app_dir}\n")
+
+# Step 1: Preload ALL src modules with proper names (no backslashes)
+print("Loading src modules:")
+src_modules = {}
+for item in app_dir.iterdir():
+    name = item.name
+    if name.startswith("src\\") and name.endswith(".py"):
+        # Extract module name: "src\models.py" -> "models"
+        module_name = name.split("\\")[-1].replace(".py", "")
+        
+        try:
+            spec = importlib.util.spec_from_file_location(module_name, str(item))
+            module = importlib.util.module_from_spec(spec)
+            sys.modules[module_name] = module
+            src_modules[module_name] = module
+            spec.loader.exec_module(module)
+            print(f"   {module_name}")
+        except Exception as e:
+            print(f"    {module_name}: {e}")
 
-# List all items - backslashes are literal in filenames!
+# Step 2: Fix imports in the code
+print("\nPreparing ui/app.py code:")
 ui_app_path = None
 for item in app_dir.iterdir():
-    print(f"  {item.name}")
-    # Look for the literal filename "ui\app.py"
     if item.name == "ui\\app.py":
         ui_app_path = item
+        break
 
 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:
+    with open(str(ui_app_path), 'r', encoding='utf-8', errors='ignore') 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 "
-    )
+    # Fix all src imports
+    original_code = code
+    code = code.replace("from src.", "from ")
+    code = code.replace("import src.", "import ")
     
-    # 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(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}")
+    print(f"   Fixed imports in ui/app.py")
     
-    # Execute the modified code
-    exec_globals = {"__name__": "__main__"}
+    # Step 3: Execute the code
+    print("\nLaunching Gradio app:")
     try:
-        exec(compile(code, str(ui_app_path), "exec"), exec_globals)
+        exec_globals = {
+            "__name__": "__main__",
+            "__file__": str(ui_app_path),
+            "create_gradio_app": None
+        }
+        exec_locals = {}
+        
+        exec(compile(code, str(ui_app_path), "exec"), exec_globals, exec_locals)
+        
+        # Get the function from locals
+        create_gradio_app = exec_globals.get("create_gradio_app") or exec_locals.get("create_gradio_app")
+        
+        if create_gradio_app:
+            demo = create_gradio_app()
+            demo.launch(
+                server_name="0.0.0.0",
+                server_port=int(os.environ.get("PORT", 7860)),
+                show_error=True
+            )
+        else:
+            print("ERROR: create_gradio_app not found after execution")
+            
     except Exception as e:
-        print(f"Error executing ui/app.py: {e}")
+        print(f"ERROR: {e}")
         import traceback
         traceback.print_exc()
 else:
-    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}")
+    print(f"ERROR: ui\\app.py not found")