Update app.py

app.py

@@ -24,7 +24,7 @@ from prompts import SYSTEM_PROMPT
 # NEW: For Q&A
 from qa import transcribe_audio_deepgram, handle_qa_exchange
 
-MAX_QA_QUESTIONS = 5
+MAX_QA_QUESTIONS = 5  # up to 5 voice/text questions
 
 def parse_user_edited_transcript(edited_text: str, host_name: str, guest_name: str):
     """
@@ -68,10 +68,8 @@ def parse_user_edited_transcript(edited_text: str, host_name: str, guest_name: s
             text=text_line
         )
         items.append(item)
-
     return items
 
-
 def regenerate_audio_from_dialogue(dialogue_items, custom_bg_music_path=None):
     """
     Re-generates multi-speaker audio from user-edited DialogueItems,
@@ -83,7 +81,6 @@ def regenerate_audio_from_dialogue(dialogue_items, custom_bg_music_path=None):
     crossfade_duration = 50  # ms
 
     for item in dialogue_items:
-        # Generate TTS for each line
         audio_file = generate_audio_mp3(item.text, item.speaker)
         seg = AudioSegment.from_file(audio_file, format="mp3")
         audio_segments.append(seg)
@@ -109,7 +106,6 @@ def regenerate_audio_from_dialogue(dialogue_items, custom_bg_music_path=None):
 
     return audio_bytes, transcript
 
-
 def generate_podcast(
     file,
     url,
@@ -169,6 +165,7 @@ def generate_podcast(
         except Exception as e:
             return None, f"Error researching topic: {str(e)}"
 
+    from utils import truncate_text
     text = truncate_text(text)
 
     extra_instructions = []
@@ -185,13 +182,13 @@ def generate_podcast(
             f"Sponsor Content Provided (should be under ~30 seconds):\n{sponsor_content}"
         )
 
-    from prompts import SYSTEM_PROMPT  # Ensure we have the main system prompt
+    from prompts import SYSTEM_PROMPT
     combined_instructions = "\n\n".join(extra_instructions).strip()
     full_prompt = SYSTEM_PROMPT
     if combined_instructions:
         full_prompt += f"\n\n# Additional Instructions\n{combined_instructions}\n"
 
-    from utils import generate_script  # We'll call our main script generator
+    from utils import generate_script, generate_audio_mp3, mix_with_bg_music
     try:
         script = generate_script(
             full_prompt,
@@ -209,7 +206,6 @@ def generate_podcast(
     transcript = ""
     crossfade_duration = 50
 
-    from utils import generate_audio_mp3, mix_with_bg_music  # Just for clarity
     try:
         for item in script.dialogue:
             audio_file = generate_audio_mp3(item.text, item.speaker)
@@ -458,13 +454,16 @@ def main():
                 st.markdown("### Updated Transcript")
                 st.markdown(new_transcript)
 
-        # POST-PODCAST Q&A
+        # -----------------------
+        # POST-PODCAST Q&A Logic
+        # -----------------------
         st.markdown("## Post-Podcast Q&A")
         used_questions = st.session_state["qa_count"]
         remaining = MAX_QA_QUESTIONS - used_questions
 
         if remaining > 0:
             st.write(f"You can ask up to {remaining} more question(s).")
+
             typed_q = st.text_input("Type your follow-up question:")
             audio_q = st.file_uploader("Or upload an audio question (WAV, MP3)")