Update app.py

app.py

@@ -58,13 +58,18 @@ def process_pdf_to_podcast(pdf_file):
         f"Steps 1-3 completed. Preview and adjust the rewritten transcript if needed.", 
         text_preview,
         transcript_preview,
-        tts_ready_preview
+        tts_ready_preview,
+        session_dir 
     )
 
 
 # Final Step: Generate Audio after optional adjustments
 @spaces.GPU
-def generate_audio_from_modified_text(tts_ready_text):
+def generate_audio_from_modified_text(tts_ready_text,session_dir):
+
+    tts_ready_path = os.path.join(session_dir, "podcast_ready_data.pkl")
+    audio_output_path = os.path.join(session_dir, "final_podcast_audio.mp3")
+    
     # Save any modified TTS-ready transcript
     with open(tts_ready_path, 'wb') as f:
         pickle.dump(tts_ready_text, f)
@@ -98,18 +103,20 @@ with gr.Blocks() as app:
     # Button for generating audio with editable transcript
     generate_audio_button = gr.Button("Generate Audio from Edited Transcript")
     final_audio_output = gr.Audio(label="Generated Podcast Audio")
+
+    session_dir = gr.State() 
     
     # Step 1-3 execution
     run_all_button.click(
         process_pdf_to_podcast, 
         inputs=pdf_input, 
-        outputs=[output_status, extracted_text_preview, transcript_preview, tts_ready_preview]
+        outputs=[output_status, extracted_text_preview, transcript_preview, tts_ready_preview,session_dir]
     )
 
     # Final step: Generate Audio from modified TTS-ready transcript
     generate_audio_button.click(
         generate_audio_from_modified_text, 
-        inputs=tts_ready_preview, 
+        inputs=[tts_ready_preview, session_dir]
         outputs=[output_status, final_audio_output]
     )