Update app.py

app.py

@@ -86,8 +86,9 @@ def select_relevant_chunk(question, chunks, chunk_embeds):
 def _chat_display_to_messages(chat_display):
     msgs = []
     for user, assistant in chat_display:
-        msgs.append({"role": "user", "content": user})
-        msgs.append({"role": "assistant", "content": assistant})
+        if user and assistant:
+            msgs.append({"role": "user", "content": user})
+            msgs.append({"role": "assistant", "content": assistant})
     return msgs
 
 # ------------------ Transcription & LLM ------------------
@@ -226,27 +227,15 @@ def generate_pdf_file(text, filename_prefix="summary"):
     pdf.output(file_path)
     return file_path
 
-def download_pdf_summary(session_id):
-    summary_text = "\n".join([m["content"] for m in SESSION_HISTORY.get(session_id, []) if m["role"]=="assistant"])
-    if not summary_text:
-        summary_text = "No summary available."
-    return generate_pdf_file(summary_text, "summary")
-
-# ------------------ Full summary including Voice, PDF, Image ------------------
+# ------------------ Full summary download (SAFE) ------------------
 def download_full_summary(session_voice, session_pdf, session_image):
     combined = []
 
-    for m in SESSION_HISTORY.get(session_voice, []):
-        if m["role"] == "assistant":
-            combined.append("🎤 VOICE:\n" + m["content"])
-
-    for m in SESSION_HISTORY.get(session_pdf, []):
-        if m["role"] == "assistant":
-            combined.append("📄 PDF:\n" + m["content"])
-
-    for m in SESSION_HISTORY.get(session_image, []):
-        if m["role"] == "assistant":
-            combined.append("🖼 IMAGE:\n" + m["content"])
+    for session, label in zip([session_voice, session_pdf, session_image], ["🎤 VOICE", "📄 PDF", "🖼 IMAGE"]):
+        history = SESSION_HISTORY.get(session, [])
+        for m in history:
+            if isinstance(m, dict) and m.get("role") == "assistant" and isinstance(m.get("content"), str):
+                combined.append(f"{label}:\n{m['content']}")
 
     if not combined:
         combined.append("No summary available.")
@@ -254,7 +243,7 @@ def download_full_summary(session_voice, session_pdf, session_image):
     summary_text = "\n\n".join(combined)
     return generate_pdf_file(summary_text, "full_summary")
 
-# ------------------ Voice & Chat Handlers (FIXED: no repeated text) ------------------
+# ------------------ Voice & Chat Handlers ------------------
 def _append_chat_display(session_id, user_text, assistant_text):
     if session_id not in CHAT_DISPLAY:
         CHAT_DISPLAY[session_id] = []
@@ -352,7 +341,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         btn_image.click(fn=handle_voice_image, inputs=[mic, session_image, tts_lang], outputs=[answer_voice, audio_output, chat_voice])
         btn_reset_logs.click(lambda: (str(uuid.uuid4()), [], None, None, ""), outputs=[session_voice, chat_voice, mic, audio_output, answer_voice])
 
-        # FIXED: Full summary download
+        # FIXED: Full summary download safely
         btn_download_logs.click(
             download_full_summary,
             inputs=[session_voice, session_pdf, session_image],