Update app.py

app.py

@@ -53,7 +53,7 @@ def process_screenplay(pdf_file, progress=gr.Progress()):
             cleaned_text = f.read()
             
         progress(1.0, desc="Complete!")
-        return [cleaned_text, gr.update(interactive=True, variant="primary"), 
+        return [cleaned_text, str(cleaned_path), gr.update(interactive=True, variant="primary"), 
                 gr.update(interactive=True, variant="primary"), 
                 gr.update(interactive=False, variant="secondary"),
                 gr.update(interactive=False, variant="secondary"),
@@ -77,11 +77,12 @@ def generate_coverage(progress=gr.Progress()):
         if not success:
             raise gr.Error("Failed to generate coverage")
             
-        with open(Path("coverage.txt"), 'r') as f:
+        coverage_path = Path("coverage.txt")
+        with open(coverage_path, 'r') as f:
             coverage = f.read()
             
         progress(1.0, desc="Complete!")
-        return [coverage, console.get_output()]
+        return [coverage, str(coverage_path), console.get_output()]
         
     except Exception as e:
         error_msg = f"Error: {str(e)}"
@@ -101,11 +102,12 @@ def analyze_screenplay(progress=gr.Progress()):
         if not success:
             raise gr.Error("Failed to generate creative analysis")
             
-        with open(Path("creative_analysis.txt"), 'r') as f:
+        analysis_path = Path("creative_analysis.txt")
+        with open(analysis_path, 'r') as f:
             analysis = f.read()
             
         progress(1.0, desc="Complete!")
-        return [analysis, gr.update(interactive=True, variant="primary"), console.get_output()]
+        return [analysis, str(analysis_path), gr.update(interactive=True, variant="primary"), console.get_output()]
         
     except Exception as e:
         error_msg = f"Error: {str(e)}"
@@ -130,7 +132,7 @@ def post_process_analysis(progress=gr.Progress()):
             analysis = f.read()
             
         progress(1.0, desc="Complete!")
-        return [analysis, gr.update(interactive=True, variant="primary"), console.get_output()]
+        return [analysis, str(output_path), gr.update(interactive=True, variant="primary"), console.get_output()]
         
     except Exception as e:
         error_msg = f"Error: {str(e)}"
@@ -151,11 +153,12 @@ def generate_market_analysis(progress=gr.Progress()):
         if not success:
             raise gr.Error("Failed to generate market analysis")
             
-        with open(Path("market_analysis.txt"), 'r') as f:
+        market_path = Path("market_analysis.txt")
+        with open(market_path, 'r') as f:
             analysis = f.read()
             
         progress(1.0, desc="Complete!")
-        return [analysis, console.get_output()]
+        return [analysis, str(market_path), console.get_output()]
         
     except Exception as e:
         error_msg = f"Error: {str(e)}"
@@ -183,56 +186,56 @@ with gr.Blocks(title="Screenplay Coverage Generator") as demo:
             with gr.Row():
                 cleaned_output = gr.Textbox(label="Cleaned Screenplay", lines=10, show_copy_button=True)
             with gr.Row():
-                gr.File(value="cleaned_screenplay_long.txt", label="Download Cleaned Screenplay")
+                cleaned_file = gr.File(label="Download Cleaned Screenplay", interactive=True, visible=False)
                 
         with gr.TabItem("Coverage"):
             with gr.Row():
                 coverage_output = gr.Textbox(label="Coverage Document", lines=10, show_copy_button=True)
             with gr.Row():
-                gr.File(value="coverage.txt", label="Download Coverage")
+                coverage_file = gr.File(label="Download Coverage", interactive=True, visible=False)
                 
         with gr.TabItem("Creative Analysis"):
             with gr.Row():
                 analysis_output = gr.Textbox(label="Creative Analysis", lines=10, show_copy_button=True)
             with gr.Row():
-                gr.File(value="creative_analysis.txt", label="Download Creative Analysis")
+                analysis_file = gr.File(label="Download Creative Analysis", interactive=True, visible=False)
                 
         with gr.TabItem("Post-Processed Analysis"):
             with gr.Row():
                 post_process_output = gr.Textbox(label="Post-Processed Analysis", lines=10, show_copy_button=True)
             with gr.Row():
-                gr.File(value="cleaned_creative_analysis.txt", label="Download Post-Processed Analysis")
+                post_process_file = gr.File(label="Download Post-Processed Analysis", interactive=True, visible=False)
                 
         with gr.TabItem("Market Analysis"):
             with gr.Row():
                 market_output = gr.Textbox(label="Market Analysis", lines=10, show_copy_button=True)
             with gr.Row():
-                gr.File(value="market_analysis.txt", label="Download Market Analysis")
+                market_file = gr.File(label="Download Market Analysis", interactive=True, visible=False)
     
     process_btn.click(
         fn=process_screenplay,
         inputs=[file_input],
-        outputs=[cleaned_output, coverage_btn, analysis_btn, post_process_btn, market_btn, console]
+        outputs=[cleaned_output, cleaned_file, coverage_btn, analysis_btn, post_process_btn, market_btn, console]
     )
     
     coverage_btn.click(
         fn=generate_coverage,
-        outputs=[coverage_output, console]
+        outputs=[coverage_output, coverage_file, console]
     )
     
     analysis_btn.click(
         fn=analyze_screenplay,
-        outputs=[analysis_output, post_process_btn, console]
+        outputs=[analysis_output, analysis_file, post_process_btn, console]
     )
     
     post_process_btn.click(
         fn=post_process_analysis,
-        outputs=[post_process_output, market_btn, console]
+        outputs=[post_process_output, post_process_file, market_btn, console]
     )
     
     market_btn.click(
         fn=generate_market_analysis,
-        outputs=[market_output, console]
+        outputs=[market_output, market_file, console]
     )
 
 if __name__ == "__main__":