fixed static file paths

app.py

@@ -32,6 +32,7 @@ import markdown
 import PyPDF2
 import io
 import copy
+from pathlib import Path
 
 def get_instructor_name(speaker):
     instructor_names = {
@@ -54,14 +55,16 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Set up environment
-OUTPUT_DIR = os.path.join(os.getcwd(), "outputs")
-UPLOAD_DIR = os.path.join(os.getcwd(), "uploads")
+# Set up environment with absolute paths
+OUTPUT_DIR = os.path.abspath(os.path.join(os.getcwd(), "outputs"))
+UPLOAD_DIR = os.path.abspath(os.path.join(os.getcwd(), "uploads"))
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 logger.info(f"Using output directory: {OUTPUT_DIR}")
 logger.info(f"Using upload directory: {UPLOAD_DIR}")
-os.environ["COQUI_TOS_AGREED"] = "1"
+
+# Set up static paths for Gradio
+gr.set_static_paths(paths=[Path(OUTPUT_DIR).absolute()])
 
 # Initialize TTS model
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -406,16 +409,13 @@ def create_zip_of_files(file_paths):
 
 # Access local files
 def get_gradio_file_url(local_path):
+    if not local_path:
+        return None
+    # Get the relative path from the current working directory
     relative_path = os.path.relpath(local_path, os.getcwd())
-    print(f"Relative path: {relative_path}")
-    
-    # Handle audio files differently for Hugging Face deployment
-    if relative_path.endswith(('.mp3', '.wav')):
-        # For audio files, use the static file serving
-        return f"/file/{relative_path}"
-    
-    # For other files, use the gradio API
-    return f"/gradio_api/file={relative_path}"
+    final_path = f"/gradio_api/file={relative_path}"
+    print(f"File URL path: {final_path}")
+    return final_path
 
 # Async generate lecture materials and audio
 async def on_generate(api_service, api_key, serpapi_key, title, lecture_content_description, lecture_type, lecture_style, speaker_audio, num_slides):
@@ -430,29 +430,20 @@ async def on_generate(api_service, api_key, serpapi_key, title, lecture_content_
     instructor_name = get_instructor_name(speaker)
     logger.info(f"Using instructor: {instructor_name}")
 
-    # Clear the outputs directory with detailed logging
     if os.path.exists(OUTPUT_DIR):
         try:
-            logger.info(f"Starting to clear outputs directory: {OUTPUT_DIR}")
-            cleared_files = []
             for item in os.listdir(OUTPUT_DIR):
                 item_path = os.path.join(OUTPUT_DIR, item)
-                try:
-                    if os.path.isfile(item_path):
-                        os.unlink(item_path)
-                        cleared_files.append(f"Deleted file: {item}")
-                    elif os.path.isdir(item_path):
-                        shutil.rmtree(item_path)
-                        cleared_files.append(f"Deleted directory: {item}")
-                except Exception as e:
-                    logger.error(f"Failed to delete {item_path}: {str(e)}")
-            logger.info(f"Successfully cleared outputs directory. Deleted items: {', '.join(cleared_files)}")
+                if os.path.isfile(item_path):
+                    os.unlink(item_path)
+                elif os.path.isdir(item_path):
+                    shutil.rmtree(item_path)
+            logger.info("Cleared outputs directory: %s", OUTPUT_DIR)
         except Exception as e:
-            logger.error(f"Failed to clear outputs directory: {str(e)}")
-            raise RuntimeError(f"Failed to clear outputs directory: {str(e)}")
+            logger.error("Failed to clear outputs directory: %s", str(e))
     else:
         os.makedirs(OUTPUT_DIR, exist_ok=True)
-        logger.info(f"Created outputs directory: {OUTPUT_DIR}")
+        logger.info("Created outputs directory: %s", OUTPUT_DIR)
     
     # Total slides include user-specified content slides plus Introduction and Closing slides
     content_slides = num_slides
@@ -841,7 +832,7 @@ Example: 'Received {total_slides} slides, {total_slides} scripts, and HTML files
         for i in range(len(scripts)):
             audio_timeline += f'<audio id="audio-{i+1}" controls src="" style="display: inline-block; margin: 0 10px; width: 200px;"><span>Loading...</span></audio>'
         
-        file_paths = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(('.md', '.txt', '.mp3', '.wav'))]
+        file_paths = [f for f in os.listdir(OUTPUT_DIR) if f.endswith(('.md', '.txt'))]
         file_paths.sort()
         file_paths = [os.path.join(OUTPUT_DIR, f) for f in file_paths]
         
@@ -939,9 +930,7 @@ Example: 'Received {total_slides} slides, {total_slides} scripts, and HTML files
         audio_timeline = ""
         for j, url in enumerate(audio_urls):
             if url:
-                # Ensure audio URL is properly formatted for Hugging Face deployment
-                audio_url = get_gradio_file_url(url)
-                audio_timeline += f'<audio id="audio-{j+1}" controls src="{audio_url}" style="display: inline-block; margin: 0 10px; width: 200px;"></audio>'
+                audio_timeline += f'<audio id="audio-{j+1}" controls src="{url}" style="display: inline-block; margin: 0 10px; width: 200px;"></audio>'
             else:
                 audio_timeline += f'<audio id="audio-{j+1}" controls src="" style="display: inline-block; margin: 0 10px; width: 200px;"><span>Audio unavailable</span></audio>'