Update app.py

app.py

@@ -1,35 +1,172 @@
 """
-PDF Text Extractor Application
-Main entry point for the PDF Text Extractor application.
+Interface creation module for PDF Text Extractor.
+Defines the Gradio interface components and layout.
 """
 
-import os
-from dotenv import load_dotenv
-from ui import create_interface
-from utils.config import check_api_key, get_app_config
+import gradio as gr
+from pdf_text_extractor import PDFTextExtractor
+from ui.handlers import copy_text, download_text, process_images_for_display
+from ui.components import (
+    create_header, create_upload_section, create_action_button,
+    create_text_display, create_action_buttons, create_image_gallery, apply_custom_css
+)
+from gradio_pdf import PDF  # <-- Added import
 
-def main():
-    """Main function to launch the application."""
+def create_dummy_interface() -> gr.Blocks:
+    """
+    Create a simple interface for when the API key is not configured.
     
-    # Load environment variables from .env file
-    load_dotenv()
+    Returns:
+        gr.Blocks: Gradio interface with disabled functionality
+    """
+    with gr.Blocks(title="PDF Text Extractor") as interface:
+        gr.Markdown("""
+        # 🔍 PDF Text Extractor
+        
+        ⚠️ **API key not configured.** Please set MISTRAL_API_KEY environment variable and restart the application.
+        """)
+        
+        with gr.Row():
+            gr.File(label="Upload PDF", file_types=[".pdf"])
+        
+        with gr.Row():
+            gr.Button("Extract Text", variant="primary", interactive=False)
+        
+        with gr.Row():
+            gr.Textbox(
+                label="Extracted Text", 
+                lines=10,
+                value="API key not configured. Text extraction is unavailable.",
+                interactive=False
+            )
+        
+        with gr.Row():
+            gr.Textbox(
+                label="Status", 
+                lines=2,
+                value="❌ MISTRAL_API_KEY environment variable is not set. Please set it and restart the application."
+            )
+        
+        with gr.Row():
+            gr.Button("📋 Copy to Clipboard", interactive=False)
+            gr.Button("📥 Download as Text File", interactive=False)
     
-    # Check for API key
-    check_api_key()
+    return interface
+
+
+
+def create_main_interface(extractor: PDFTextExtractor) -> gr.Blocks:
+    """
+    Create the main application interface.
     
-    # Create and launch the interface
-    interface = create_interface()
+    Args:
+        extractor: PDFTextExtractor instance
+        
+    Returns:
+        gr.Blocks: Gradio interface with full functionality    """
+    # Make the extractor a local function attribute
+    def process_pdf_wrapper(pdf_file):
+        """Process PDF with the extractor from closure"""
+        extracted_text, status, images_data = extractor.extract_text_from_pdf(pdf_file)
+        # Get PDF file path for image extraction
+        pdf_path = pdf_file.name if hasattr(pdf_file, 'name') else pdf_file if pdf_file else None
+        gallery_images = process_images_for_display(images_data, pdf_path)
+        return extracted_text, status, gallery_images
     
-    # Get application configuration
-    app_config = get_app_config()
-      
-    # Launch with appropriate settings
-    interface.launch(
-        # server_port=app_config["server_port"],
-        debug=app_config["debug"],
-        quiet=app_config["quiet"],
-        max_file_size=app_config["max_file_size"]
-    )
+    with gr.Blocks(title="🔍 PDF Text Extractor", theme=gr.themes.Soft()) as interface:
+        create_header()
+        # Add file upload section
+        with gr.Row():
+            pdf_input = create_upload_section()
+        # Add PDF viewer for uploaded file
+        with gr.Row():
+            pdf_viewer = PDF(label="PDF Preview", height=500)
+        # Add extract button
+        with gr.Row():
+            submit_btn = create_action_button()
+        # Add status display
+        with gr.Row():
+            status_output = gr.Textbox(
+                label="Status", 
+                lines=2,
+                placeholder="Upload a PDF to see status..."
+            )
+        # Create tabs for text and images
+        with gr.Tabs():
+            with gr.TabItem("Extracted Text"):
+                text_output = gr.Textbox(
+                    label="Extracted Text", 
+                    lines=15,
+                    max_lines=30,
+                    placeholder="Extracted text will appear here...",
+                    show_copy_button=True
+                )
+                with gr.Row():
+                    copy_btn, download_btn = create_action_buttons()
+            with gr.TabItem("Extracted Images"):
+                image_gallery = create_image_gallery()
+                image_info = gr.Markdown("Images extracted from the PDF will appear here.")
+        # Set up function calls
+        # Update PDF viewer when a file is uploaded
+        pdf_input.change(
+            fn=lambda f: f,
+            inputs=pdf_input,
+            outputs=pdf_viewer
+        )
+        submit_btn.click(
+            fn=process_pdf_wrapper,
+            inputs=[pdf_input],
+            outputs=[text_output, status_output, image_gallery]
+        )
+        copy_btn.click(
+            fn=copy_text,
+            inputs=text_output,
+            outputs=None,
+            js="""
+            function(text) {
+                if (text) {
+                    navigator.clipboard.writeText(text);
+                    // Show a temporary notification
+                    var notification = document.createElement('div');
+                    notification.textContent = 'Text copied to clipboard!';
+                    notification.style.position = 'fixed';
+                    notification.style.bottom = '20px';
+                    notification.style.left = '50%';
+                    notification.style.transform = 'translateX(-50%)';
+                    notification.style.padding = '10px 20px';
+                    notification.style.background = '#4CAF50';
+                    notification.style.color = 'white';
+                    notification.style.borderRadius = '4px';
+                    notification.style.zIndex = '1000';
+                    document.body.appendChild(notification);
+                    setTimeout(function() {
+                        document.body.removeChild(notification);
+                    }, 2000);
+                }
+                return text;
+            }
+            """
+        )
+        download_btn.click(
+            fn=download_text,
+            inputs=text_output,
+            outputs=gr.File(label="Download", elem_id="download_output"),
+            show_progress=False
+        )
+        apply_custom_css()
+    return interface
 
-if __name__ == "__main__":
-    main()
+def create_interface() -> gr.Blocks:
+    """
+    Create and configure the Gradio interface.
+    
+    Returns:
+        gr.Blocks: Configured Gradio interface
+    """
+    # Initialize the PDF extractor
+    try:
+        extractor = PDFTextExtractor()
+        return create_main_interface(extractor)
+    except ValueError as e:
+        # Create a dummy interface if API key is missing
+        return create_dummy_interface()