Update app.py

app.py

@@ -1,52 +1,15 @@
 import os
 import gradio as gr
 import tempfile
-from video_processing import (
-    process_video_file,
-)
-from quiz_processing_1 import analyze_document
+from src.mainFunctions import *
+
 ELEVENLABS_API_KEY = os.environ.get("ELEVENLABS_API_KEY", None)
 GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY', None)
+CLAUDE_API_KEY = os.environ.get('CLAUDE_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 = analyze_document(
-        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.Blocks(title="Document to Quiz Generator") as app:
+    gr.Markdown("# Document & Media => Quiz")
+    gr.Markdown("Upload a video, audio, PDF, or Word document to automatically generate a quiz with topics, key concepts, summaries, and questions.")
     
     with gr.Row():
         with gr.Column():
@@ -61,13 +24,18 @@ with gr.Blocks(title="Video to Quiz Generator") as app:
                 value="scribe_v1",
                 label="Transcription Model"
             )
-            google_api_key = gr.Textbox(
-                placeholder="Enter your Google Gemini or Claude or OpenAI API key",
-                label="Google API Key (for quiz generation)",
+            gemini_api_key = gr.Textbox(
+                placeholder="Enter your Google Gemini API key",
+                label="Google Gemini API Key",
                 type="password",
+                value=GOOGLE_API_KEY
             )
-            
-    # Add the new course information textboxes
+            claude_api_key = gr.Textbox(
+                placeholder="Enter your Claude API key",
+                label="Claude API Key",
+                type="password"
+            )
+
     with gr.Row():
         with gr.Column():
             course_name = gr.Textbox(
@@ -82,7 +50,15 @@ with gr.Blocks(title="Video to Quiz Generator") as app:
                 placeholder="Enter the lesson name",
                 label="Lesson Name"
             )
-            
+
+    with gr.Row():
+        with gr.Column():
+            language_selector = gr.Radio(
+                choices=["Uzbek", "English", "Russian"],
+                value="English",
+                label="Content Language"
+            )
+
     with gr.Tabs():
         with gr.TabItem("Upload Video"):
             with gr.Row():
@@ -104,29 +80,152 @@ with gr.Blocks(title="Video to Quiz Generator") as app:
                     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
-                ]
-            )
+        
+        # PDF Tab
+        with gr.TabItem("Upload PDF"):
+            with gr.Row():
+                with gr.Column():
+                    pdf_input = gr.File(label="Upload PDF", file_types=[".pdf"])
+                    process_pdf_button = gr.Button("Process PDF & Generate Quiz")
+                with gr.Column():
+                    pdf_status_output = gr.Textbox(label="PDF Processing Status")
+                    pdf_text_file_output = gr.File(label="Extracted Text File")
+            with gr.Row():
+                with gr.Column():
+                    pdf_quiz_output = gr.Textbox(
+                        label="Generated Quiz",
+                        lines=15
+                    )
+                    with gr.Row():
+                        pdf_quiz_file_output = gr.File(label="Download Quiz Text")
+                        pdf_json_file_output = gr.File(label="Download Quiz JSON")
+        
+        # Word Document Tab
+        with gr.TabItem("Upload Word Document"):
+            with gr.Row():
+                with gr.Column():
+                    docx_input = gr.File(label="Upload Word Document", file_types=[".docx"])
+                    process_docx_button = gr.Button("Process Word Document & Generate Quiz")
+                with gr.Column():
+                    docx_status_output = gr.Textbox(label="Word Document Processing Status")
+                    docx_text_file_output = gr.File(label="Extracted Text File")
+            with gr.Row():
+                with gr.Column():
+                    docx_quiz_output = gr.Textbox(
+                        label="Generated Quiz",
+                        lines=15
+                    )
+                    with gr.Row():
+                        docx_quiz_file_output = gr.File(label="Download Quiz Text")
+                        docx_json_file_output = gr.File(label="Download Quiz JSON")
+        
+        # Audio Tab
+        with gr.TabItem("Upload Audio"):
+            with gr.Row():
+                with gr.Column():
+                    audio_input = gr.Audio(label="Upload Audio", type="filepath")
+                    process_audio_button = gr.Button("Process Audio & Generate Quiz")
+                with gr.Column():
+                    audio_status_output = gr.Textbox(label="Audio Processing Status")
+                    audio_transcript_file_output = gr.File(label="Transcription Text File")
+            with gr.Row():
+                with gr.Column():
+                    audio_quiz_output = gr.Textbox(
+                        label="Generated Quiz",
+                        lines=15
+                    )
+                    with gr.Row():
+                        audio_quiz_file_output = gr.File(label="Download Quiz Text")
+                        audio_json_file_output = gr.File(label="Download Quiz JSON")
+
+    # Connect video processing
+    extract_button_file.click(
+        fn=combined_process_video_file,
+        inputs=[
+            video_input, 
+            format_choice_file, 
+            elevenlabs_api_key, 
+            model_id, 
+            gemini_api_key,
+            claude_api_key,
+            course_name,
+            section_name,
+            lesson_name,
+            language_selector 
+        ],
+        outputs=[
+            audio_output_file, 
+            status_output_file, 
+            transcript_file_output,
+            transcript_status_output, 
+            quiz_output_file, 
+            quiz_file_output_file, 
+            json_file_output_file
+        ]
+    )
+    
+    # Connect PDF processing
+    process_pdf_button.click(
+        fn=process_pdf_document,
+        inputs=[
+            pdf_input,
+            gemini_api_key,
+            claude_api_key,
+            course_name,
+            section_name,
+            lesson_name,
+            language_selector
+        ],
+        outputs=[
+            pdf_status_output,
+            pdf_text_file_output,
+            pdf_quiz_output,
+            pdf_quiz_file_output,
+            pdf_json_file_output
+        ]
+    )
+    
+    process_docx_button.click(
+        fn=process_docx_document,
+        inputs=[
+            docx_input,
+            gemini_api_key,
+            claude_api_key,
+            course_name,
+            section_name,
+            lesson_name,
+            language_selector
+        ],
+        outputs=[
+            docx_status_output,
+            docx_text_file_output,
+            docx_quiz_output,
+            docx_quiz_file_output,
+            docx_json_file_output
+        ]
+    )
+    
+    process_audio_button.click(
+        fn=process_audio_document,
+        inputs=[
+            audio_input,
+            elevenlabs_api_key,
+            model_id,
+            gemini_api_key,
+            claude_api_key,
+            course_name,
+            section_name,
+            lesson_name,
+            language_selector
+        ],
+        outputs=[
+            audio_status_output,
+            audio_transcript_file_output,
+            audio_quiz_output,
+            audio_quiz_file_output,
+            audio_json_file_output
+        ]
+    )
 
 if __name__ == "__main__":
     app.launch(share=True, debug=True)