Update app.py

app.py

@@ -4,6 +4,7 @@ from src.processing.gemini_processor import GeminiProcessor
 from src.analysis.coverage_generator import CoverageGenerator
 from src.analysis.creative_analyzer import CreativeAnalyzer
 from src.analysis.analysis_post_processor import AnalysisPostProcessor
+from src.analysis.market_analyzer import MarketAnalyzer
 from pathlib import Path
 import logging
 
@@ -55,6 +56,7 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
         return [cleaned_text, gr.update(interactive=True, variant="primary"), 
                 gr.update(interactive=True, variant="primary"), 
                 gr.update(interactive=False, variant="secondary"),
+                gr.update(interactive=False, variant="secondary"),
                 console.get_output()]
 
     except Exception as e:
@@ -127,6 +129,31 @@ def post_process_analysis(progress=gr.Progress()):
         with open(output_path, '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 generate_market_analysis(progress=gr.Progress()):
+    console = ConsoleOutput()
+    logger = setup_logging(console)
+    
+    try:
+        market_analyzer = MarketAnalyzer()
+        progress(0.5, desc="Generating market analysis...")
+        coverage_path = Path("coverage.txt")
+        creative_analysis_path = Path("cleaned_creative_analysis.txt")
+        success = market_analyzer.generate_market_analysis(coverage_path, creative_analysis_path)
+        
+        if not success:
+            raise gr.Error("Failed to generate market analysis")
+            
+        with open(Path("market_analysis.txt"), 'r') as f:
+            analysis = f.read()
+            
         progress(1.0, desc="Complete!")
         return [analysis, console.get_output()]
         
@@ -146,6 +173,7 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
         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)
+        market_btn = gr.Button("Market Analysis", interactive=False)
     
     with gr.Row():
         console = gr.Textbox(label="Console Output", lines=10, max_lines=30, autoscroll=True, show_copy_button=True)
@@ -159,11 +187,13 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
             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)
+        with gr.TabItem("Market Analysis"):
+            market_output = gr.Textbox(label="Market Analysis", lines=10, show_copy_button=True)
     
     process_btn.click(
         fn=process_screenplay,
         inputs=[file_input],
-        outputs=[cleaned_output, coverage_btn, analysis_btn, post_process_btn, console]
+        outputs=[cleaned_output, coverage_btn, analysis_btn, post_process_btn, market_btn, console]
     )
     
     coverage_btn.click(
@@ -178,7 +208,12 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
     
     post_process_btn.click(
         fn=post_process_analysis,
-        outputs=[post_process_output, console]
+        outputs=[post_process_output, market_btn, console]
+    )
+    
+    market_btn.click(
+        fn=generate_market_analysis,
+        outputs=[market_output, console]
     )
 
 if __name__ == "__main__":