Update gradio_app.py

gradio_app.py

@@ -3,6 +3,8 @@ import gradio as gr
 import asyncio
 import uuid
 import threading
+import subprocess
+import shutil
 from datetime import datetime
 import logging
 import traceback
@@ -71,7 +73,6 @@ def delete_job(job_id):
             try:
                 # Remove the entire output directory for this job
                 output_dir = os.path.dirname(job['output_file'])
-                import shutil
                 shutil.rmtree(output_dir, ignore_errors=True)
             except Exception as e:
                 logger.error(f"Error removing output files: {e}")
@@ -722,29 +723,92 @@ with gr.Blocks(
                         )
     
     with gr.Tab("📂 Job History & Management"):
-        with gr.Row():
-            with gr.Column(scale=3):
-                refresh_jobs_btn = gr.Button("🔄 Refresh Job List", variant="secondary")
-            with gr.Column(scale=1):
-                clear_completed_btn = gr.Button("🧹 Clear Completed Jobs", variant="secondary")
-                clear_all_btn = gr.Button("🗑️ Clear All Jobs", variant="stop")
-        
+        # Job list table (full width)
         jobs_table = gr.Dataframe(
             headers=["ID", "Topic", "Status", "Progress (%)", "Start Time", "Message"],
             datatype=["str", "str", "str", "number", "str", "str"],
             interactive=False,
-            label="📋 Job History",
-            wrap=True
+            label=None,
+            wrap=True,
+            elem_classes=["job-history-table"]
         )
-        
+        # Action buttons (horizontal row, full width)
         with gr.Row():
-            with gr.Column():
-                select_job_btn = gr.Button("👁️ View Selected Job", variant="primary")
-                selected_job_id = gr.Textbox(label="Selected Job ID", visible=False)
-            with gr.Column():
-                delete_job_btn = gr.Button("🗑️ Delete Selected Job", variant="stop")
-                download_job_btn = gr.Button("💾 Download Job Results", variant="secondary")
-    
+            select_job_btn = gr.Button("👁️ View Details", variant="primary", size="sm")
+            delete_job_btn = gr.Button("🗑️ Delete", variant="stop", size="sm")
+            download_job_btn = gr.Button("💾 Download", variant="secondary", size="sm")
+            refresh_jobs_btn = gr.Button("🔄 Refresh List", variant="secondary", size="sm")
+            clear_completed_btn = gr.Button("🧹 Clear Completed", variant="secondary", size="sm")
+            clear_all_btn = gr.Button("🗑️ Clear All", variant="stop", size="sm")
+        selected_job_id = gr.Textbox(label="Selected Job ID", visible=False)
+        # Job details viewer (full width, below buttons)
+        with gr.Group(elem_classes=["job-details-panel"]):
+            gr.Markdown("""
+            <div style='font-size:1.2em; font-weight:600; margin-bottom:0.5em;'>
+                👁️ <span style='color:#3b82f6'>Job Details Viewer</span>
+            </div>
+            """)
+            close_details_btn = gr.Button("⬅️ Back to Job List", variant="secondary", size="sm", visible=False)
+            job_details_container = gr.Column(visible=False)
+            with job_details_container:
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        job_topic_display = gr.Textbox(label="📚 Topic", interactive=False)
+                        job_description_display = gr.Textbox(label="📝 Description", interactive=False, lines=3)
+                        job_model_display = gr.Textbox(label="🤖 Model Used", interactive=False)
+                    with gr.Column(scale=1):
+                        job_status_display = gr.Textbox(label="📊 Status", interactive=False)
+                        job_progress_display = gr.Number(label="📈 Progress (%)", interactive=False)
+                        job_start_time_display = gr.Textbox(label="⏰ Start Time", interactive=False)
+                with gr.Row():
+                    job_processing_time_display = gr.Textbox(label="⏱️ Processing Time", interactive=False)
+                    job_message_display = gr.Textbox(label="💬 Current Message", interactive=False)
+                with gr.Column(visible=False) as job_video_container:
+                    gr.Markdown("### 🎬 Generated Video")
+                    job_video_player = gr.Video(
+                        label="Video Output", 
+                        interactive=False,
+                        show_download_button=True,
+                        height=300
+                    )
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            job_thumbnail_display = gr.Image(
+                                label="🖼️ Thumbnail", 
+                                height=150,
+                                interactive=False
+                            )
+                        with gr.Column(scale=2):
+                            job_scene_gallery = gr.Gallery(
+                                label="🎨 Scene Previews", 
+                                columns=3, 
+                                object_fit="contain", 
+                                height=150,
+                                show_download_button=True
+                            )
+                with gr.Column(visible=False) as job_error_container:
+                    gr.Markdown("### ❌ Error Details")
+                    job_error_display = gr.Textbox(
+                        label="Error Message", 
+                        interactive=False, 
+                        lines=3
+                    )
+                    job_stack_trace_display = gr.Textbox(
+                        label="Stack Trace", 
+                        interactive=False, 
+                        lines=5,
+                        max_lines=10
+                    )
+            no_job_selected = gr.Markdown(
+            """
+            <div style='padding:2em 0;text-align:center;color:#888;'>
+                <b>📄 No Job Selected</b><br>
+                Select a job from the list to view its details.
+            </div>
+            """,
+            visible=True
+            )
+
     with gr.Tab("ℹ️ Help & Documentation"):
         gr.Markdown("""
         ## 🎯 How to Use Theory2Manim
@@ -1027,6 +1091,7 @@ with gr.Blocks(
     )
     
     # Job history tab functions
+
     def load_job_list():
         jobs = get_job_list()
         rows = []
@@ -1040,33 +1105,111 @@ with gr.Blocks(
                     formatted_time = start_time
             else:
                 formatted_time = 'Unknown'
-            
             rows.append([
-                job['id'][:8] + '...', 
-                job['topic'][:50] + ('...' if len(job['topic']) > 50 else ''), 
-                job['status'].title(), 
-                job['progress'], 
+                job['id'][:8] + '...',
+                job['topic'][:50] + ('...' if len(job['topic']) > 50 else ''),
+                job['status'].title(),
+                job['progress'],
                 formatted_time,
                 job['message'][:100] + ('...' if len(job['message']) > 100 else '')
             ])
         return rows
     
     def select_job(evt: gr.EventData):
-        if not evt:
-            return "", "No job selected"
-        
-        selected_row = evt.index[0] if hasattr(evt, 'index') and evt.index else 0
+        if not evt or not hasattr(evt, 'index') or not evt.index:
+            # No job selected
+            return "", "No job selected", gr.update(visible=False)
+        selected_row = evt.index[0]
         jobs = get_job_list()
         if selected_row < len(jobs):
-            return jobs[selected_row]['id'], f"Selected job: {jobs[selected_row]['topic']}"
-        return "", "No job selected"
+            # Job selected
+            return jobs[selected_row]['id'], f"Selected job: {jobs[selected_row]['topic']}", gr.update(visible=True)
+        return "", "No job selected", gr.update(visible=False)
     
+    def back_to_job_list():
+        # Show job list, hide details
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+
+    def view_job_details(job_id):
+        """View details of a selected job."""
+        if not job_id or job_id not in job_status:
+            # Return 17 outputs, all hidden or empty
+            return (
+                gr.update(visible=False),  # job_details_container
+                gr.update(visible=True),   # no_job_selected
+                "", "", "", "", 0, "", "", "",  # topic, desc, model, status, progress, start, proc_time, msg
+                gr.update(visible=False),  # job_video_container
+                gr.update(visible=False, value=None),  # job_video_player
+                gr.update(visible=False, value=None),  # job_thumbnail_display
+                gr.update(visible=False, value=[]),    # job_scene_gallery
+                gr.update(visible=False),  # job_error_container
+                gr.update(visible=False, value=""),   # job_error_display
+                gr.update(visible=False, value="")    # job_stack_trace_display
+            )
+        job = job_status[job_id]
+        # Format start time
+        start_time = job.get('start_time', '')
+        if start_time:
+            try:
+                dt = datetime.fromisoformat(start_time.replace('Z', '+00:00'))
+                formatted_time = dt.strftime('%Y-%m-%d %H:%M:%S')
+            except:
+                formatted_time = start_time
+        else:
+            formatted_time = 'Unknown'
+        
+        # Video and error visibility
+        is_completed = job.get('status') == 'completed'
+        is_failed = job.get('status') == 'failed'
+        # Always return 17 outputs in order
+        return (
+            gr.update(visible=True),  # job_details_container
+            gr.update(visible=False),  # no_job_selected
+            job.get('topic', ''),
+            job.get('description', ''),
+            job.get('model', ''),
+            gr.update(value=job.get('status', '').title()),  # status_display
+            job.get('progress', 0),
+            formatted_time,
+            job.get('processing_time', ''),
+            job.get('message', ''),
+            gr.update(visible=is_completed),  # job_video_container
+            gr.update(visible=is_completed, value=job.get('output_file') if is_completed else None),  # job_video_player
+            gr.update(visible=is_completed and job.get('thumbnail') is not None, value=job.get('thumbnail') if is_completed else None),  # job_thumbnail_display
+            gr.update(visible=is_completed, value=job.get('scene_snapshots', []) if is_completed else []),  # job_scene_gallery
+            gr.update(visible=is_failed),  # job_error_container
+            gr.update(visible=is_failed, value=job.get('error', '') if is_failed else ""),  # job_error_display
+            gr.update(visible=is_failed, value=job.get('stack_trace', '') if is_failed else "")  # job_stack_trace_display
+        )
+
     def delete_selected_job(job_id):
-        if job_id:
-            result = delete_job(job_id)
-            return result, ""
-        return "No job selected", ""
+        """Delete the selected job and update the UI."""
+        if not job_id or job_id not in job_status:
+            return "Job not found", None, gr.update(visible=False)
+        
+        # Delete the job
+        result = delete_job(job_id)
+        
+        # Update job list
+        jobs = get_job_list()
+        
+        # Refresh job table
+        return result, gr.update(value=load_job_list()), gr.update(visible=False)
+
+    def download_job_results(job_id):
+        """Download the results of a job."""
+        if not job_id or job_id not in job_status:
+            return "Job not found", None
+        
+        job = job_status[job_id]
+        output_file = job.get('output_file')
+        
+        if not output_file or not os.path.exists(output_file):
+            return "Output file not found", None
+        
+        return "Download started", output_file
     
+    # Connect job history tab event handlers
     refresh_jobs_btn.click(
         fn=load_job_list,
         outputs=[jobs_table]
@@ -1077,17 +1220,31 @@ with gr.Blocks(
     
     jobs_table.select(
         fn=select_job,
-        outputs=[selected_job_id, result_text]
+        outputs=[selected_job_id, result_text, close_details_btn]
     )
     
     select_job_btn.click(
-        fn=lambda x: gr.update(visible=True) if x else gr.update(visible=False),
+        fn=view_job_details,
         inputs=[selected_job_id],
-        outputs=[status_container]
-    ).then(
-        fn=update_status_display,
+        outputs=[
+            job_details_container, no_job_selected,
+            job_topic_display, job_description_display, job_model_display,
+            job_status_display, job_progress_display, job_start_time_display,
+            job_processing_time_display, job_message_display,
+            job_video_container, job_video_player, job_thumbnail_display, job_scene_gallery,
+            job_error_container, job_error_display, job_stack_trace_display
+        ]
+    )
+    
+    close_details_btn.click(
+        fn=back_to_job_list,
+        outputs=[job_details_container, no_job_selected, close_details_btn]
+    )
+    
+    download_job_btn.click(
+        fn=download_job_results,
         inputs=[selected_job_id],
-        outputs=[status_text, video_output, video_output, thumbnail_preview, scene_gallery, processing_time_text, cancel_btn]
+        outputs=[result_text]
     )
     
     delete_job_btn.click(
@@ -1145,6 +1302,8 @@ with gr.Blocks(
     )
 
 
+
+
 if __name__ == "__main__":
     import os
     app.queue().launch(