Update app.py

app.py

@@ -1,19 +1,52 @@
 import gradio as gr
-import google.generativeai as genai
 import os
+from src.processing.gemini_processor import GeminiProcessor
+from src.analysis.coverage_generator import CoverageGenerator
+from pathlib import Path
 
-def process_pdf(pdf_file):
-    # For initial testing
-    return "PDF processed", "Coverage generated"
+def process_pdf(pdf_file, progress=gr.Progress()):
+    if pdf_file is None:
+        raise gr.Error("Please upload a PDF file")
+        
+    progress(0.1, desc="Initializing...")
+    processor = GeminiProcessor()
+    coverage_gen = CoverageGenerator()
+    
+    try:
+        # Clean screenplay
+        progress(0.4, desc="Processing screenplay...")
+        cleaned_path = Path("cleaned_screenplay_long.txt")
+        success = processor.process_screenplay(pdf_file.name, str(cleaned_path))
+        if not success:
+            raise gr.Error("Failed to process screenplay")
+            
+        # Generate coverage
+        progress(0.8, desc="Generating coverage...")
+        success = coverage_gen.generate_coverage(cleaned_path)
+        if not success:
+            raise gr.Error("Failed to generate coverage")
+            
+        # Read results
+        with open(cleaned_path, 'r') as f:
+            cleaned_text = f.read()
+        with open(Path("coverage.txt"), 'r') as f:
+            coverage = f.read()
+            
+        progress(1.0, desc="Complete!")
+        return cleaned_text, coverage
+        
+    except Exception as e:
+        raise gr.Error(f"Error: {str(e)}")
 
 demo = gr.Interface(
     fn=process_pdf,
     inputs=gr.File(label="Upload Screenplay PDF", file_types=[".pdf"]),
     outputs=[
-        gr.Textbox(label="Cleaned Text"),
-        gr.Textbox(label="Coverage")
+        gr.Textbox(label="Cleaned Screenplay", lines=10),
+        gr.Textbox(label="Coverage Document", lines=10)
     ],
-    title="Screenplay Coverage Generator"
+    title="Screenplay Coverage Generator",
+    description="Upload a screenplay PDF to generate coverage analysis"
 )
 
 if __name__ == "__main__":