Fix directory case handling and improve module imports

Dockerfile

@@ -45,15 +45,16 @@ RUN pip install --no-cache-dir openmim && \
 
 # Clone and set up MuseV with exact case
 RUN git clone https://github.com/TMElyralab/MuseV.git && \
+    mv MuseV musev_temp && \
+    mv musev_temp MuseV && \
     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 .. && \
-    # 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
+    cp -r scripts/gradio/* ../ && \
+    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"
@@ -66,7 +67,8 @@ RUN echo "Current directory:" && pwd && \
     echo "Directory contents:" && ls -la && \
     echo "MuseV directory contents:" && ls -la MuseV && \
     echo "Gradio scripts directory contents:" && ls -la MuseV/scripts/gradio && \
-    echo "Python path:" && echo $PYTHONPATH
+    echo "Python path:" && echo $PYTHONPATH && \
+    echo "Checking gradio scripts:" && ls -la gradio_*.py
 
 EXPOSE 7860
 

app.py

@@ -16,7 +16,16 @@ 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")  # Note the capital V
+
+# Look for MuseV directory with either case
+if os.path.exists(os.path.join(ProjectDir, "MuseV")):
+    MuseVDir = os.path.join(ProjectDir, "MuseV")
+elif os.path.exists(os.path.join(ProjectDir, "musev")):
+    MuseVDir = os.path.join(ProjectDir, "musev")
+else:
+    print("Error: Could not find MuseV directory")
+    sys.exit(1)
+
 GradioScriptsDir = os.path.join(MuseVDir, "scripts", "gradio")
 
 print(f"Project directory: {ProjectDir}")
@@ -65,7 +74,7 @@ for path in sys.path:
 
 print("Attempting to import gradio modules...")
 
-# First check if the files exist
+# First check if the files exist in the current directory
 print("Checking for gradio module files...")
 video2video_path = os.path.join(ProjectDir, "gradio_video2video.py")
 text2video_path = os.path.join(ProjectDir, "gradio_text2video.py")
@@ -82,36 +91,33 @@ if os.path.exists(text2video_path):
 else:
     print(f"Warning: text2video not found at {text2video_path}")
 
+# Try importing from current directory first
 try:
-    from gradio_video2video import online_v2v_inference
-    print("Successfully imported video2video")
+    import gradio_video2video
+    import gradio_text2video
+    online_v2v_inference = gradio_video2video.online_v2v_inference
+    online_t2v_inference = gradio_text2video.online_t2v_inference
+    print("Successfully imported gradio modules from current directory")
 except Exception as e:
-    print(f"Error importing video2video: {str(e)}")
-    print(f"Current directory: {os.getcwd()}")
-    print(f"Directory contents: {os.listdir('.')}")
-    if os.path.exists(GradioScriptsDir):
-        print(f"Gradio scripts directory contents: {os.listdir(GradioScriptsDir)}")
-
-try:
-    from gradio_text2video import online_t2v_inference
-    print("Successfully imported text2video")
-except Exception as e:
-    print(f"Error importing text2video: {str(e)}")
-    print("Attempting to import directly from MuseV...")
+    print(f"Error importing from current directory: {str(e)}")
+    print("Attempting to import from MuseV directory...")
     try:
         sys.path.insert(0, os.path.join(MuseVDir, "scripts", "gradio"))
+        from gradio_video2video import online_v2v_inference
         from gradio_text2video import online_t2v_inference
-        print("Successfully imported text2video from MuseV directory")
+        print("Successfully imported from MuseV directory")
     except Exception as e2:
         print(f"Error importing from MuseV directory: {str(e2)}")
+        print("Current directory contents:")
+        print(os.listdir('.'))
+        if os.path.exists(GradioScriptsDir):
+            print("Gradio scripts directory contents:")
+            print(os.listdir(GradioScriptsDir))
+        sys.exit(1)
 
 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,