Fix file copying and directory handling

Dockerfile

@@ -43,7 +43,10 @@ RUN pip install --no-cache-dir openmim && \
     mim install "mmdet>=3.1.0" && \
     mim install "mmpose>=1.1.0"
 
-# Clone and set up MuseV first
+# Copy application code first (excluding MuseV directory)
+COPY --chown=user:user README.md requirements.txt app.py ./
+
+# Clone and set up MuseV
 RUN git clone https://github.com/TMElyralab/MuseV.git && \
     cd MuseV && \
     git clone https://github.com/huggingface/diffusers.git && \
@@ -56,9 +59,6 @@ RUN git clone https://github.com/TMElyralab/MuseV.git && \
     cp scripts/gradio/gradio_text2video.py ../gradio_text2video.py && \
     chmod +x ../gradio_*.py
 
-# Now copy application code
-COPY --chown=user:user . .
-
 # 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

@@ -66,6 +66,8 @@ if musev_candidates:
             print("Successfully renamed musev to MuseV")
         except Exception as e:
             print(f"Warning: Could not rename directory: {str(e)}")
+            # If we can't rename, use the existing directory
+            MuseVDir = os.path.join(ProjectDir, "musev")
 else:
     print("Warning: Could not find MuseV directory")
     sys.exit(1)
@@ -117,32 +119,31 @@ for path in sys.path:
 
 print("Attempting to import gradio modules...")
 
-# Search for gradio modules in multiple locations
-video2video_path = find_file("gradio_video2video.py", ProjectDir)
-text2video_path = find_file("gradio_text2video.py", ProjectDir)
+# First try to find modules in current directory
+video2video_path = os.path.join(ProjectDir, "gradio_video2video.py")
+text2video_path = os.path.join(ProjectDir, "gradio_text2video.py")
 
 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}")
-        
-        # If found in MuseV, copy to app directory
-        if video2video_path and text2video_path:
-            shutil.copy2(video2video_path, ProjectDir)
-            shutil.copy2(text2video_path, ProjectDir)
-            video2video_path = os.path.join(ProjectDir, "gradio_video2video.py")
-            text2video_path = os.path.join(ProjectDir, "gradio_text2video.py")
-            print("Copied gradio scripts to app directory")
-
-if not video2video_path or not text2video_path:
-    print("Error: Could not find gradio modules")
-    sys.exit(1)
+# If not found in current directory, look in MuseV directory
+if not os.path.exists(video2video_path) or not os.path.exists(text2video_path):
+    print("Modules not found in current directory, checking MuseV directory...")
+    musev_video2video = os.path.join(GradioScriptsDir, "gradio_video2video.py")
+    musev_text2video = os.path.join(GradioScriptsDir, "gradio_text2video.py")
+    
+    if os.path.exists(musev_video2video) and os.path.exists(musev_text2video):
+        print("Found modules in MuseV directory, copying to current directory...")
+        shutil.copy2(musev_video2video, video2video_path)
+        shutil.copy2(musev_text2video, text2video_path)
+        print("Successfully copied modules")
+    else:
+        print("Error: Could not find modules in MuseV directory")
+        print(f"MuseV scripts directory contents:")
+        if os.path.exists(GradioScriptsDir):
+            print(os.listdir(GradioScriptsDir))
+        sys.exit(1)
 
 try:
     print("Attempting to import modules...")