Update app.py

app.py

@@ -4,6 +4,7 @@ import glob
 import re
 import pandas as pd
 import gradio as gr
+import spaces  # <--- REQUIRED FOR HUGGING FACE ZEROGPU TIERS
 from pypdf import PdfReader
 import docx2txt
 import speech_recognition as sr
@@ -21,6 +22,7 @@ os.makedirs(PROCESSED_DIR, exist_ok=True)
 # -------------------------------------------------------------
 # CORE PIPELINE LOGIC (TRANSCRIBER, PARSER, CHUNKER)
 # -------------------------------------------------------------
+@spaces.GPU  # <--- TELLS HUGGING FACE TO ALLOCATE GPU POWER FOR TRANSCRIBING
 def transcribe_video_audio(file_path):
     try:
         gr.Info(f"🎬 Extracting track layers from {os.path.basename(file_path)}...")
@@ -102,6 +104,7 @@ def structure_unsloth_rows(chunks, archetype):
             rows.append({"text": chunk})
     return rows
 
+@spaces.GPU  # <--- DECORATES THE TOP LEVEL CONVERSION Pipeline FOR ZEROGPU STABILITY
 def execute_dataset_builder_pipeline(files, archetype, enable_chunking, chunk_size, chunk_overlap):
     if not files:
         return "⚠️ Target file loading queue is empty. Please upload files."
@@ -141,12 +144,10 @@ def execute_dataset_builder_pipeline(files, archetype, enable_chunking, chunk_si
 # THEME TOGGLE SWITCH LOGIC ☀️/🌙
 # -------------------------------------------------------------
 def toggle_theme(current_theme):
-    # This toggles the utility class directly via Gradio's internal state mapping
     if current_theme == "dark":
         return gr.update(variant="light"), "light"
     return gr.update(variant="dark"), "dark"
 
-# CUSTOM JAVASCRIPT FOR THE THEME SWITCHING IN GRADIO RUNTIME
 js_theme_switcher = """
 function(theme) {
     const documentElement = document.documentElement;
@@ -168,15 +169,14 @@ archetype_choices = [
     "📖 Raw Knowledge Base (Continued Pre-Training)"
 ]
 
-# Base Theme Configurations
 custom_theme = gr.themes.Default(
     primary_hue="green",
     secondary_hue="zinc",
     neutral_hue="zinc"
 )
 
-with gr.Blocks(theme=custom_theme, title="Unsloth Studio Platform") as demo:
-    # State keeping track of the user UI layout status
+# Removed theme and title parameters from constructor to prevent Gradio 6 layout deprecation warnings
+with gr.Blocks() as demo:
     ui_theme_state = gr.State("dark")
     
     with gr.Row():
@@ -200,14 +200,12 @@ with gr.Blocks(theme=custom_theme, title="Unsloth Studio Platform") as demo:
         with gr.Column(scale=1):
             log_monitor = gr.Textbox(label="🖥️ Core Engine Pipeline Logs", lines=15)
             
-    # Trigger execution conversion functionality
     run_btn.click(
         fn=execute_dataset_builder_pipeline,
         inputs=[file_uploader, archetype_dropdown, chunk_toggle, size_input, overlap_input],
         outputs=log_monitor
     )
     
-    # Theme Switcher Trigger Interactions
     theme_toggle_btn.click(
         fn=toggle_theme,
         inputs=[ui_theme_state],
@@ -218,9 +216,9 @@ with gr.Blocks(theme=custom_theme, title="Unsloth Studio Platform") as demo:
         js=js_theme_switcher
     )
 
-    # Initialize into dark layout out-of-the-box
     demo.load(fn=lambda: "dark", outputs=ui_theme_state).then(fn=None, inputs=[ui_theme_state], js=js_theme_switcher)
 
 if __name__ == "__main__":
-    demo.launch()
+    # Theme configuration parameters passed to launch method matching Gradio 6 guidelines
+    demo.launch(theme=custom_theme)