Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 from src.processing.gemini_processor import GeminiProcessor
 from src.analysis.coverage_generator import CoverageGenerator
+from src.analysis.creative_analyzer import CreativeAnalyzer
 from pathlib import Path
 import logging
 
@@ -24,12 +25,6 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
     logging.basicConfig(level=logging.DEBUG)
     logger = logging.getLogger()
     logger.handlers = []
-    
-    class GradioHandler(logging.Handler):
-        def emit(self, record):
-            msg = self.format(record)
-            console.write(msg)
-            
     logger.addHandler(GradioHandler())
     
     try:
@@ -40,12 +35,13 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
         
         if not success:
             raise gr.Error("Failed to process screenplay")
-        
+            
         with open(cleaned_path, 'r') as f:
             cleaned_text = f.read()
             
         progress(1.0, desc="Complete!")
-        return [cleaned_text, gr.update(interactive=True, variant="primary"), console.get_output()]
+        return [cleaned_text, gr.update(interactive=True, variant="primary"), 
+                gr.update(interactive=True, variant="primary"), console.get_output()]
     
     except Exception as e:
         error_msg = f"Error: {str(e)}"
@@ -57,12 +53,6 @@ def generate_coverage(progress=gr.Progress()):
     logging.basicConfig(level=logging.DEBUG)
     logger = logging.getLogger()
     logger.handlers = []
-    
-    class GradioHandler(logging.Handler):
-        def emit(self, record):
-            msg = self.format(record)
-            console.write(msg)
-            
     logger.addHandler(GradioHandler())
     
     try:
@@ -85,6 +75,33 @@ def generate_coverage(progress=gr.Progress()):
         console.write(error_msg)
         raise gr.Error(error_msg)
 
+def analyze_screenplay(progress=gr.Progress()):
+    console = ConsoleOutput()
+    logging.basicConfig(level=logging.DEBUG)
+    logger = logging.getLogger()
+    logger.handlers = []
+    logger.addHandler(GradioHandler())
+    
+    try:
+        analyzer = CreativeAnalyzer()
+        progress(0.5, desc="Performing creative analysis...")
+        cleaned_path = Path("cleaned_screenplay_long.txt")
+        success = analyzer.analyze_screenplay(cleaned_path)
+        
+        if not success:
+            raise gr.Error("Failed to generate creative analysis")
+            
+        with open(Path("creative_analysis.txt"), 'r') as f:
+            analysis = f.read()
+            
+        progress(1.0, desc="Complete!")
+        return [analysis, console.get_output()]
+        
+    except Exception as e:
+        error_msg = f"Error: {str(e)}"
+        console.write(error_msg)
+        raise gr.Error(error_msg)
+
 with gr.Blocks(title="Screenplay Coverage Generator") as demo:
     gr.Markdown("# Screenplay Coverage Generator")
     
@@ -94,6 +111,7 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
     with gr.Row():
         process_btn = gr.Button("Process Screenplay", variant="primary")
         coverage_btn = gr.Button("Generate Coverage", interactive=False)
+        analysis_btn = gr.Button("Creative Analysis", interactive=False)
     
     with gr.Row():
         console = gr.Textbox(label="Console Output", lines=10, max_lines=30, autoscroll=True, show_copy_button=True)
@@ -103,17 +121,24 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
             cleaned_output = gr.Textbox(label="Cleaned Screenplay", lines=10, show_copy_button=True)
         with gr.TabItem("Coverage"):
             coverage_output = gr.Textbox(label="Coverage Document", lines=10, show_copy_button=True)
+        with gr.TabItem("Creative Analysis"):
+            analysis_output = gr.Textbox(label="Creative Analysis", lines=10, show_copy_button=True)
     
     process_btn.click(
         fn=process_screenplay,
         inputs=[file_input],
-        outputs=[cleaned_output, coverage_btn, console]
+        outputs=[cleaned_output, coverage_btn, analysis_btn, console]
     )
     
     coverage_btn.click(
         fn=generate_coverage,
         outputs=[coverage_output, console]
     )
+    
+    analysis_btn.click(
+        fn=analyze_screenplay,
+        outputs=[analysis_output, console]
+    )
 
 if __name__ == "__main__":
     demo.launch()