Update app.py

app.py

@@ -1,83 +1,133 @@
-# app.py
-import os
-import gradio as gr
-import tempfile
-from video_processing import (
-    process_video_file,
-)
-from quiz_processing import process_text
-
-ELEVENLABS_API_KEY = os.environ.get("ELEVENLABS_API_KEY", None)
-GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY', None)
-
-def combined_process_video_file(video_file, output_format, elevenlabs_api_key, model_id, google_api_key):
-    audio_path, audio_msg, transcript_file, transcript_msg, transcript_text, _ = process_video_file(
-        video_file, output_format, elevenlabs_api_key, model_id
-    )
-
-    if not transcript_text:
-        return audio_path, audio_msg, transcript_file, transcript_msg, "No transcript text to analyze", None, None
-
-    formatted_quiz, quiz_file, json_file = process_text(transcript_text, google_api_key)
-
-    return audio_path, audio_msg, transcript_file, transcript_msg, formatted_quiz, quiz_file, json_file
-
-# Create the Gradio interface
-with gr.Blocks(title="Video to Quiz Generator") as app:
-    gr.Markdown("Video => Quiz")
-    gr.Markdown("Upload a video or provide a URL to extract audio, transcribe, and automatically generate a quiz with topics, key concepts, summaries, and questions.")
-
-    with gr.Row():
-        with gr.Column():
-            elevenlabs_api_key = gr.Textbox(
-                placeholder="Enter your ElevenLabs API key",
-                label="ElevenLabs API Key (for transcription)",
-                type="password",
-                value=ELEVENLABS_API_KEY
-            )
-
-            model_id = gr.Dropdown(
-                choices=["scribe_v1"],
-                value="scribe_v1",
-                label="Transcription Model"
-            )
-
-            google_api_key = gr.Textbox(
-                placeholder="Enter your Google Gemini API key",
-                label="Google API Key (for quiz generation)",
-                type="password",
-                value=GOOGLE_API_KEY
-            )
-
-    with gr.Tabs():
-        with gr.TabItem("Upload Video"):
-            with gr.Row():
-                with gr.Column():
-                    video_input = gr.Video(label="Upload Video")
-                    format_choice_file = gr.Radio(["mp3", "wav"], value="mp3", label="Audio Format")
-                    extract_button_file = gr.Button("Process Video & Generate Quiz")
-
-                with gr.Column():
-                    audio_output_file = gr.Audio(label="Extracted Audio", type="filepath")
-                    status_output_file = gr.Textbox(label="Audio Extraction Status")
-                    transcript_file_output = gr.File(label="Transcription Text File")
-                    transcript_status_output = gr.Textbox(label="Transcription Status")
-
-            with gr.Row():
-                with gr.Column():
-                    quiz_output_file = gr.Textbox(
-                        label="Generated Quiz",
-                        lines=15
-                    )
-                    with gr.Row():
-                        quiz_file_output_file = gr.File(label="Download Quiz Text")
-                        json_file_output_file = gr.File(label="Download Quiz JSON")
-
-            extract_button_file.click(
-                fn=combined_process_video_file,
-                inputs=[video_input, format_choice_file, elevenlabs_api_key, model_id, google_api_key],
-                outputs=[audio_output_file, status_output_file, transcript_file_output,
-                         transcript_status_output, quiz_output_file, quiz_file_output_file, json_file_output_file]
-            )
-if __name__ == "__main__":
-    app.launch(share= True, debug = True)
+import os
+import gradio as gr
+import tempfile
+from video_processing import (
+    process_video_file,
+)
+from quiz_processing import process_text
+ELEVENLABS_API_KEY = os.environ.get("ELEVENLABS_API_KEY", None)
+GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY', None)
+
+def combined_process_video_file(video_file, output_format, elevenlabs_api_key, model_id, google_api_key, 
+                                course_name, section_name, lesson_name):
+    result = process_video_file(
+        video_file, output_format, elevenlabs_api_key, model_id
+    )
+    
+    audio_path = result[0]
+    audio_msg = result[1]
+    transcript_file = result[2]
+    transcript_msg = result[3]
+    transcript_text = result[4]
+    
+    if not transcript_text:
+        return audio_path, audio_msg, transcript_file, transcript_msg, "No transcript text to analyze", None, None
+    
+    enriched_transcript = f"""
+    COURSE: {course_name}
+    SECTION: {section_name}
+    LESSON: {lesson_name}
+    
+    TRANSCRIPT:
+    {transcript_text}
+    """
+    
+    # Call process_text with the course information parameters
+    formatted_quiz, quiz_file, json_file = process_text(
+        enriched_transcript, 
+        google_api_key, 
+        course_name, 
+        section_name, 
+        lesson_name
+    )
+    
+    return audio_path, audio_msg, transcript_file, transcript_msg, formatted_quiz, quiz_file, json_file
+
+
+with gr.Blocks(title="Video to Quiz Generator") as app:
+    gr.Markdown("Video => Quiz")
+    gr.Markdown("Upload a video or provide a URL to extract audio, transcribe, and automatically generate a quiz with topics, key concepts, summaries, and questions.")
+    
+    with gr.Row():
+        with gr.Column():
+            elevenlabs_api_key = gr.Textbox(
+                placeholder="Enter your ElevenLabs API key",
+                label="ElevenLabs API Key (for transcription)",
+                type="password",
+                value=ELEVENLABS_API_KEY
+            )
+            model_id = gr.Dropdown(
+                choices=["scribe_v1"],
+                value="scribe_v1",
+                label="Transcription Model"
+            )
+            google_api_key = gr.Textbox(
+                placeholder="Enter your Google Gemini API key",
+                label="Google API Key (for quiz generation)",
+                type="password",
+                value=GOOGLE_API_KEY
+            )
+            
+    # Add the new course information textboxes
+    with gr.Row():
+        with gr.Column():
+            course_name = gr.Textbox(
+                placeholder="Enter the course name",
+                label="Course Name"
+            )
+            section_name = gr.Textbox(
+                placeholder="Enter the section name",
+                label="Section Name"
+            )
+            lesson_name = gr.Textbox(
+                placeholder="Enter the lesson name",
+                label="Lesson Name"
+            )
+            
+    with gr.Tabs():
+        with gr.TabItem("Upload Video"):
+            with gr.Row():
+                with gr.Column():
+                    video_input = gr.Video(label="Upload Video")
+                    format_choice_file = gr.Radio(["mp3", "wav"], value="mp3", label="Audio Format")
+                    extract_button_file = gr.Button("Process Video & Generate Quiz")
+                with gr.Column():
+                    audio_output_file = gr.Audio(label="Extracted Audio", type="filepath")
+                    status_output_file = gr.Textbox(label="Audio Extraction Status")
+                    transcript_file_output = gr.File(label="Transcription Text File")
+                    transcript_status_output = gr.Textbox(label="Transcription Status")
+            with gr.Row():
+                with gr.Column():
+                    quiz_output_file = gr.Textbox(
+                        label="Generated Quiz",
+                        lines=15
+                    )
+                    with gr.Row():
+                        quiz_file_output_file = gr.File(label="Download Quiz Text")
+                        json_file_output_file = gr.File(label="Download Quiz JSON")
+                        
+            extract_button_file.click(
+                fn=combined_process_video_file,
+                inputs=[
+                    video_input, 
+                    format_choice_file, 
+                    elevenlabs_api_key, 
+                    model_id, 
+                    google_api_key,
+                    course_name,
+                    section_name,
+                    lesson_name
+                ],
+                outputs=[
+                    audio_output_file, 
+                    status_output_file, 
+                    transcript_file_output,
+                    transcript_status_output, 
+                    quiz_output_file, 
+                    quiz_file_output_file, 
+                    json_file_output_file
+                ]
+            )
+
+if __name__ == "__main__":
+    app.launch(share=True, debug=True)