more ui fixes:{

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 = {
@@ -406,9 +407,10 @@ def create_zip_of_files(file_paths):
 
 # Access local files
 def get_gradio_file_url(local_path):
+    if not local_path:
+        return None
     relative_path = os.path.relpath(local_path, os.getcwd())
-    final_path = f"/gradio_api/file={relative_path}"
-    return final_path
+    return f"/gradio_api/file={relative_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):
@@ -667,7 +669,7 @@ Example: 'Received {total_slides} slides, {total_slides} scripts, and HTML files
             elif source == "research_agent" and isinstance(message, TextMessage) and "handoff_to_slide_agent" in message.content:
                 logger.info("Research Agent completed research")
                 progress = 25
-                label = "Slides: generating..."
+                label = "Research complete. Generating slides..."
                 yield (
                     html_with_progress(label, progress),
                     []
@@ -730,7 +732,7 @@ Example: 'Received {total_slides} slides, {total_slides} scripts, and HTML files
                         except Exception as e:
                             logger.error("Error saving script to %s: %s", script_file, str(e))
                     progress = 75
-                    label = "Scripts generated and saved. Reviewing..."
+                    label = "Script complete. Reviewing lecture materials..."
                     yield (
                         html_with_progress(label, progress),
                         []
@@ -1002,12 +1004,36 @@ js_code = """
         }, interval);
     }
 
+    // Function to check if a file exists
+    async function checkFileExists(url) {
+        try {
+            const response = await fetch(url, { method: 'HEAD' });
+            if (response.ok) {
+                console.log(`File exists: ${url}`);
+                return true;
+            }
+            // Fallback: Some servers disallow HEAD, try GET request
+            if (response.status === 405 || response.status === 403) {
+                const getResp = await fetch(url, { method: 'GET' });
+                if (getResp.ok) {
+                    console.log(`File exists (GET fallback): ${url}`);
+                    return true;
+                }
+            }
+            console.log(`File not found: ${url}`);
+            return false;
+        } catch (error) {
+            console.error(`Error checking file:`, error);
+            return false;
+        }
+    }
+
     // Main initialization function
     function initializeSlides() {
         console.log("Initializing slides...");
 
         // Wait for lecture-data to load the JSON data
-        waitForElement('#lecture-data', (dataElement) => {
+        waitForElement('#lecture-data', async (dataElement) => {
             if (!dataElement.textContent) {
                 console.error("Lecture data element is empty");
                 return;
@@ -1036,15 +1062,29 @@ js_code = """
             let isReloading = false;
 
             // Wait for slide-content element
-            waitForElement('#slide-content', (slideContent) => {
+            waitForElement('#slide-content', async (slideContent) => {
                 console.log("Slide content element found");
 
                 // Initialize audio elements
                 for (let i = 0; i < totalSlides; i++) {
                     const audio = document.getElementById(`audio-${i+1}`);
                     if (audio) {
+                        const url = lectureData.audioFiles[i];
+                        if (url) {
+                            const exists = await checkFileExists(url);
+                            if (exists) {
+                                audio.src = url;
+                                audio.load();
+                                console.log(`Set audio source for audio-${i+1}:`, url);
+                            } else {
+                                console.error(`Audio file not found: ${url}`);
+                                audio.innerHTML = "<span>Audio unavailable</span>";
+                            }
+                        } else {
+                            console.log(`No audio URL for slide ${i+1}`);
+                            audio.innerHTML = "<span>No audio</span>";
+                        }
                         audioElements.push(audio);
-                        console.log(`Found audio element audio-${i+1}:`, audio);
                     } else {
                         console.error(`Audio element audio-${i+1} not found`);
                     }
@@ -1545,6 +1585,9 @@ with gr.Blocks(
     js=js_code,
     head='<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css">'
 ) as demo:
+    # Set static paths for accessing output files
+    gr.set_static_paths(paths=[Path(OUTPUT_DIR).absolute()])
+    
     gr.Markdown("""
                 ## <center>Lectūra: Your AI Genie for Self-taught Mastery.</center>
                 ### <center>(Disclaimer: This demo is part of a submission to the AgentX – LLM Agents MOOC Competition, hosted by Berkeley RDI. © Lectūra Labs. All rights reserved)</center>
@@ -2223,10 +2266,15 @@ with gr.Blocks(
             "audience": audience_val or "University"
         }
 
-    def show_note_content(evt: dict, notes):
+    def show_note_content(evt: dict, notes=None):
         # Handle both event dict and direct index
-        idx = evt.get('index', 0) if isinstance(evt, dict) else evt
-        if 0 <= idx < len(notes):
+        if isinstance(evt, dict):
+            idx = evt.get('index', 0)
+        else:
+            idx = evt
+            notes = evt  # If only one argument is provided, use it as notes
+            
+        if notes and 0 <= idx < len(notes):
             note = notes[idx]
             note_file = os.path.join(OUTPUT_DIR, f"{note['title']}.txt")
             if os.path.exists(note_file):