Improve file handling and add more debugging

Dockerfile

@@ -43,26 +43,25 @@ RUN pip install --no-cache-dir openmim && \
     mim install "mmdet>=3.1.0" && \
     mim install "mmpose>=1.1.0"
 
+# Copy application code first
+COPY --chown=user:user . .
+
 # Clone and set up MuseV
-RUN git clone https://github.com/TMElyralab/MuseV.git /tmp/MuseV && \
-    mkdir -p MuseV && \
-    cp -r /tmp/MuseV/* MuseV/ && \
+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 .. && \
-    cp -r scripts/gradio/gradio_video2video.py ../gradio_video2video.py && \
-    cp -r scripts/gradio/gradio_text2video.py ../gradio_text2video.py && \
+    # Copy gradio scripts to app directory
+    cp scripts/gradio/gradio_video2video.py ../gradio_video2video.py && \
+    cp scripts/gradio/gradio_text2video.py ../gradio_text2video.py && \
     chmod +x ../gradio_*.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"
 
-# Copy application code
-COPY --chown=user:user . .
-
 # Debug information
 RUN echo "Current directory:" && pwd && \
     echo "Directory contents:" && ls -la && \
@@ -71,8 +70,9 @@ RUN echo "Current directory:" && pwd && \
     echo "Python path:" && echo $PYTHONPATH && \
     echo "Checking gradio scripts:" && ls -la gradio_*.py && \
     echo "Verifying file contents:" && \
-    head -n 1 gradio_video2video.py && \
-    head -n 1 gradio_text2video.py
+    cat gradio_video2video.py | head -n 5 && \
+    echo "---" && \
+    cat gradio_text2video.py | head -n 5
 
 EXPOSE 7860
 

app.py

@@ -9,17 +9,33 @@ import gradio as gr
 import spaces
 import cuid
 import importlib.util
+import glob
 
 from huggingface_hub import snapshot_download
 
 print("Starting application...")
 
+def find_file(pattern, search_path="."):
+    """Find a file matching the pattern in the search path."""
+    matches = glob.glob(os.path.join(search_path, pattern))
+    if matches:
+        return matches[0]
+    return None
+
 def import_from_path(module_name, file_path):
     """Import a module from a file path."""
     if not os.path.exists(file_path):
         raise ImportError(f"File not found: {file_path}")
     
     print(f"Importing {module_name} from {file_path}")
+    with open(file_path, 'r') as f:
+        print(f"File contents (first 5 lines):")
+        for i, line in enumerate(f):
+            if i < 5:
+                print(line.strip())
+            else:
+                break
+    
     spec = importlib.util.spec_from_file_location(module_name, file_path)
     module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(module)
@@ -28,10 +44,22 @@ def import_from_path(module_name, file_path):
 # Set up paths
 ProjectDir = os.path.dirname(os.path.abspath(__file__))
 CheckpointsDir = os.path.join(ProjectDir, "checkpoints")
-MuseVDir = os.path.join(ProjectDir, "MuseV")  # We ensure this is capitalized in Dockerfile
-GradioScriptsDir = os.path.join(MuseVDir, "scripts", "gradio")
+MuseVDir = os.path.join(ProjectDir, "MuseV")
 
 print(f"Project directory: {ProjectDir}")
+print(f"Directory contents:")
+print(os.listdir(ProjectDir))
+
+# Find MuseV directory with case-insensitive search
+musev_candidates = [d for d in os.listdir(ProjectDir) if d.lower() == "musev"]
+if musev_candidates:
+    MuseVDir = os.path.join(ProjectDir, musev_candidates[0])
+    print(f"Found MuseV directory: {MuseVDir}")
+else:
+    print("Warning: Could not find MuseV directory")
+
+GradioScriptsDir = os.path.join(MuseVDir, "scripts", "gradio")
+
 print(f"MuseV directory: {MuseVDir}")
 print(f"Gradio scripts directory: {GradioScriptsDir}")
 
@@ -77,16 +105,23 @@ for path in sys.path:
 
 print("Attempting to import gradio modules...")
 
-# Try importing from current directory first
-video2video_path = os.path.join(ProjectDir, "gradio_video2video.py")
-text2video_path = os.path.join(ProjectDir, "gradio_text2video.py")
+# Search for gradio modules in multiple locations
+video2video_path = find_file("gradio_video2video.py", ProjectDir)
+text2video_path = find_file("gradio_text2video.py", ProjectDir)
 
 print(f"Looking for modules at:")
 print(f"video2video: {video2video_path}")
 print(f"text2video: {text2video_path}")
 
+if not video2video_path or not text2video_path:
+    print("Searching in MuseV directory...")
+    if os.path.exists(GradioScriptsDir):
+        video2video_path = find_file("gradio_video2video.py", GradioScriptsDir)
+        text2video_path = find_file("gradio_text2video.py", GradioScriptsDir)
+        print(f"Found in MuseV: video2video={video2video_path}, text2video={text2video_path}")
+
 try:
-    print("Attempting to import from current directory...")
+    print("Attempting to import modules...")
     video2video = import_from_path("gradio_video2video", video2video_path)
     text2video = import_from_path("gradio_text2video", text2video_path)
     online_v2v_inference = video2video.online_v2v_inference