Update app.py

app.py

@@ -1,11 +1,10 @@
-# app.py (Final version with live logging and corrections)
+# app.py (Final version with directory fix and live logging)
 
 import gradio as gr
 import uuid
 import subprocess
 import threading
 import os
-import time
 import sys
 import re
 import traceback
@@ -21,7 +20,6 @@ kokoro_voices_path = os.path.join(model_dir, "voices.bin")
 if not os.path.exists(kokoro_model_path) or not os.path.exists(kokoro_voices_path):
     print("Downloading Kokoro TTS models...")
     os.makedirs(model_dir, exist_ok=True)
-    # Using specific wget commands for clarity and robustness
     os.system(f"wget -O {kokoro_model_path} https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/kokoro-v0_19.onnx")
     os.system(f"wget -O {kokoro_voices_path} https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files/voices.bin")
     print("Model download complete.")
@@ -38,21 +36,19 @@ def run_video_generation(task_id: str, topic: str, context: str, model: str):
     
     # Sanitize topic name to create a valid directory/file prefix
     file_prefix = re.sub(r'[^a-z0-9_]+', '_', topic.lower())
-    # The generate_video.py script will create this directory inside the general 'output' folder
     output_dir = os.path.join("output", file_prefix)
     
-    # --- IMPORTANT: Command points to the specific output directory for this topic ---
+    # --- THIS IS THE FIX ---
+    # The command now correctly passes the specific output directory.
     command = [
         "python", "-u", "generate_video.py", # '-u' for unbuffered output
         "--model", model,
         "--topic", topic,
         "--context", context,
-        "--output_dir", output_dir
-        # Langfuse is disabled by not including the --use_langfuse flag
+        "--output_dir", output_dir # Use the variable, not the string "output"
     ]
 
     try:
-        # Use Popen to run the process in the background and stream output
         process = subprocess.Popen(
             command,
             stdout=subprocess.PIPE,
@@ -62,17 +58,14 @@ def run_video_generation(task_id: str, topic: str, context: str, model: str):
             universal_newlines=True,
         )
         
-        # Read output line-by-line in real-time
         for line in iter(process.stdout.readline, ''):
-            print(line, end='')  # Print to Hugging Face console logs
+            print(line, end='')
             tasks[task_id]['log'] += line
         
-        process.wait() # Wait for the process to complete
+        process.wait()
         
         if process.returncode == 0:
-            # Check for the final combined video file
             final_video_path = os.path.join(output_dir, f"{file_prefix}_combined.mp4")
-            
             if os.path.exists(final_video_path):
                 tasks[task_id]['status'] = 'completed'
                 tasks[task_id]['video_path'] = final_video_path
@@ -80,11 +73,11 @@ def run_video_generation(task_id: str, topic: str, context: str, model: str):
             else:
                 tasks[task_id]['status'] = 'failed'
                 tasks[task_id]['error'] = "Script finished, but the final combined video file was not found."
-                tasks[task_id]['log'] += f"\n❌ Error: Output video not found at {final_video_path}"
+                tasks[task_id]['log'] += f"\n❌ Error: Output video not found at {final_video_path}. Check full logs."
         else:
             tasks[task_id]['status'] = 'failed'
             tasks[task_id]['error'] = f"Process failed with return code {process.returncode}."
-            tasks[task_id]['log'] += f"\n❌ Error: Process failed. Check logs for details."
+            tasks[task_id]['log'] += f"\n❌ Error: Process failed. See logs above for details."
 
     except Exception as e:
         print(f"Caught an exception: {e}")
@@ -109,7 +102,7 @@ def start_generation(topic: str, context: str, model: str):
 
 def check_status(task_id: str):
     if not task_id:
-        return "Please provide a Task ID.", None, "Please enter a Task ID above and click 'Check Status'."
+        return "Please enter a Task ID.", None, "Please enter a Task ID above and click 'Check Status'."
         
     task = tasks.get(task_id)
     if not task:
@@ -131,8 +124,7 @@ def check_status(task_id: str):
     status_message = f"🔄 Status: {status} (Model: {model})"
     return status_message, None, log
 
-# Create the Gradio interface
-with gr.Blocks(css="footer {display: none !important}", title="Theorem Explain Agent") as demo:
+with gr.Blocks(title="Theorem Explain Agent") as demo:
     gr.Markdown("# 🎓 Theorem Explain Agent: Video Generation")
     gr.Markdown("Generate educational videos explaining mathematical theorems and concepts. This may take several minutes.")
 
@@ -163,7 +155,6 @@ with gr.Blocks(css="footer {display: none !important}", title="Theorem Explain A
         video_output = gr.Video(label="Generated Video", interactive=False)
         log_display = gr.Textbox(label="Live Generation Logs", lines=15, interactive=False)
 
-    # Connect the functions to the interface
     start_button.click(
         fn=start_generation,
         inputs=[topic_input, context_input, model_input],
@@ -174,9 +165,7 @@ with gr.Blocks(css="footer {display: none !important}", title="Theorem Explain A
         fn=check_status,
         inputs=[task_id_input],
         outputs=[status_display, video_output, log_display],
-        # Every 2 seconds, poll the status
         every=2 
     )
 
-# Launch the app
 demo.launch()