Fix path handling and add more debugging

Dockerfile

@@ -43,13 +43,17 @@ RUN pip install --no-cache-dir openmim && \
     mim install "mmdet>=3.1.0" && \
     mim install "mmpose>=1.1.0"
 
-# Clone and set up MuseV
+# Clone and set up MuseV with exact case
 RUN git clone https://github.com/TMElyralab/MuseV.git && \
     cd MuseV && \
     git clone https://github.com/huggingface/diffusers.git && \
     git clone https://github.com/patrickvonplaten/controlnet_aux.git && \
     cd diffusers && pip install -e . && \
-    cd ../controlnet_aux && pip install -e .
+    cd ../controlnet_aux && pip install -e . && \
+    cd .. && \
+    # Create symbolic links for gradio scripts
+    ln -s scripts/gradio/gradio_video2video.py ../gradio_video2video.py && \
+    ln -s scripts/gradio/gradio_text2video.py ../gradio_text2video.py
 
 # Set up Python path
 ENV PYTHONPATH="${HOME}/app:${HOME}/app/MuseV:${HOME}/app/MuseV/MMCM:${HOME}/app/MuseV/diffusers/src:${HOME}/app/MuseV/controlnet_aux/src:${HOME}/app/MuseV/scripts/gradio"

app.py

@@ -11,26 +11,35 @@ import cuid
 
 from huggingface_hub import snapshot_download
 
+print("Starting application...")
+
 # Set up paths
 ProjectDir = os.path.dirname(os.path.abspath(__file__))
 CheckpointsDir = os.path.join(ProjectDir, "checkpoints")
-MuseVDir = os.path.join(ProjectDir, "MuseV")
+MuseVDir = os.path.join(ProjectDir, "MuseV")  # Note the capital V
 GradioScriptsDir = os.path.join(MuseVDir, "scripts", "gradio")
 
+print(f"Project directory: {ProjectDir}")
+print(f"MuseV directory: {MuseVDir}")
+print(f"Gradio scripts directory: {GradioScriptsDir}")
+
 # Add the MuseV paths to sys.path
 paths_to_add = [
+    ProjectDir,  # Add current directory first
     GradioScriptsDir,
     MuseVDir,
     os.path.join(MuseVDir, "MMCM"),
     os.path.join(MuseVDir, "diffusers", "src"),
-    os.path.join(MuseVDir, "controlnet_aux", "src"),
-    os.path.dirname(os.path.abspath(__file__))
+    os.path.join(MuseVDir, "controlnet_aux", "src")
 ]
 
 for path in paths_to_add:
-    if os.path.exists(path) and path not in sys.path:
-        sys.path.insert(0, path)
-        print(f"Added {path} to PYTHONPATH")
+    if os.path.exists(path):
+        if path not in sys.path:
+            sys.path.insert(0, path)
+            print(f"Added {path} to PYTHONPATH")
+    else:
+        print(f"Warning: Path does not exist: {path}")
 
 def download_model():
     if not os.path.exists(CheckpointsDir):
@@ -55,6 +64,24 @@ for path in sys.path:
     print(f"Path: {path}")
 
 print("Attempting to import gradio modules...")
+
+# First check if the files exist
+print("Checking for gradio module files...")
+video2video_path = os.path.join(ProjectDir, "gradio_video2video.py")
+text2video_path = os.path.join(ProjectDir, "gradio_text2video.py")
+print(f"Looking for video2video at: {video2video_path}")
+print(f"Looking for text2video at: {text2video_path}")
+
+if os.path.exists(video2video_path):
+    print(f"Found video2video at {video2video_path}")
+else:
+    print(f"Warning: video2video not found at {video2video_path}")
+
+if os.path.exists(text2video_path):
+    print(f"Found text2video at {text2video_path}")
+else:
+    print(f"Warning: text2video not found at {text2video_path}")
+
 try:
     from gradio_video2video import online_v2v_inference
     print("Successfully imported video2video")
@@ -70,10 +97,21 @@ try:
     print("Successfully imported text2video")
 except Exception as e:
     print(f"Error importing text2video: {str(e)}")
+    print("Attempting to import directly from MuseV...")
+    try:
+        sys.path.insert(0, os.path.join(MuseVDir, "scripts", "gradio"))
+        from gradio_text2video import online_t2v_inference
+        print("Successfully imported text2video from MuseV directory")
+    except Exception as e2:
+        print(f"Error importing from MuseV directory: {str(e2)}")
 
 ignore_video2video = False
 max_image_edge = 1280
 
+if 'online_t2v_inference' not in locals():
+    print("ERROR: Failed to import online_t2v_inference. Cannot continue.")
+    sys.exit(1)
+
 print("Setting up Gradio interface...")
 demo = gr.Interface(
     fn=online_t2v_inference,