Update app.py

app.py

@@ -53,8 +53,10 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
             
         progress(1.0, desc="Complete!")
         return [cleaned_text, gr.update(interactive=True, variant="primary"), 
-                gr.update(interactive=True, variant="primary"), console.get_output()]
-    
+                gr.update(interactive=True, variant="primary"), 
+                gr.update(interactive=False, variant="secondary"),
+                console.get_output()]
+
     except Exception as e:
         error_msg = f"Error: {str(e)}"
         console.write(error_msg)
@@ -89,18 +91,32 @@ def analyze_screenplay(progress=gr.Progress()):
     logger = setup_logging(console)
     
     try:
-        # Step 1: Creative Analysis
         analyzer = CreativeAnalyzer()
-        progress(0.3, desc="Performing creative analysis...")
+        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")
             
-        # Step 2: Post Processing
-        progress(0.6, desc="Post-processing analysis...")
+        with open(Path("creative_analysis.txt"), 'r') as f:
+            analysis = f.read()
+            
+        progress(1.0, desc="Complete!")
+        return [analysis, gr.update(interactive=True, variant="primary"), console.get_output()]
+        
+    except Exception as e:
+        error_msg = f"Error: {str(e)}"
+        console.write(error_msg)
+        raise gr.Error(error_msg)
+
+def post_process_analysis(progress=gr.Progress()):
+    console = ConsoleOutput()
+    logger = setup_logging(console)
+    
+    try:
         post_processor = AnalysisPostProcessor()
+        progress(0.5, desc="Post-processing analysis...")
         input_path = Path("creative_analysis.txt")
         output_path = Path("cleaned_creative_analysis.txt")
         success = post_processor.process_analysis(str(input_path), str(output_path))
@@ -114,7 +130,6 @@ def analyze_screenplay(progress=gr.Progress()):
         progress(1.0, desc="Complete!")
         return [analysis, console.get_output()]
         
-        
     except Exception as e:
         error_msg = f"Error: {str(e)}"
         console.write(error_msg)
@@ -130,6 +145,7 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
         process_btn = gr.Button("Process Screenplay", variant="primary")
         coverage_btn = gr.Button("Generate Coverage", interactive=False)
         analysis_btn = gr.Button("Creative Analysis", interactive=False)
+        post_process_btn = gr.Button("Post-Process Analysis", interactive=False)
     
     with gr.Row():
         console = gr.Textbox(label="Console Output", lines=10, max_lines=30, autoscroll=True, show_copy_button=True)
@@ -141,11 +157,13 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
             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)
+        with gr.TabItem("Post-Processed Analysis"):
+            post_process_output = gr.Textbox(label="Post-Processed Analysis", lines=10, show_copy_button=True)
     
     process_btn.click(
         fn=process_screenplay,
         inputs=[file_input],
-        outputs=[cleaned_output, coverage_btn, analysis_btn, console]
+        outputs=[cleaned_output, coverage_btn, analysis_btn, post_process_btn, console]
     )
     
     coverage_btn.click(
@@ -155,7 +173,12 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
     
     analysis_btn.click(
         fn=analyze_screenplay,
-        outputs=[analysis_output, console]
+        outputs=[analysis_output, post_process_btn, console]
+    )
+    
+    post_process_btn.click(
+        fn=post_process_analysis,
+        outputs=[post_process_output, console]
     )
 
 if __name__ == "__main__":