Update app.py

app.py

@@ -159,7 +159,6 @@ def handle_pdf_question(question, session_id):
         {"role": "user", "content": f"PDF chunk:\n{chunk}\n\nQuestion: {question}"}
     ]
     assistant_text = groq_chat_completion(messages)
-    # Add snippet highlighting for wow factor
     assistant_text = f"**Snippet from PDF:**\n{chunk[:200]}...\n\n**Answer:**\n{assistant_text}"
     if session_id not in SESSION_HISTORY:
         SESSION_HISTORY[session_id] = []
@@ -233,6 +232,31 @@ def download_pdf_summary(session_id):
         summary_text = "No summary available."
     return generate_pdf_file(summary_text, "summary")
 
+# ------------------ NEW: Full merged summary ------------------
+def download_full_summary(session_voice, session_pdf, session_image):
+    combined = []
+
+    # Voice
+    for m in SESSION_HISTORY.get(session_voice, []):
+        if m["role"] == "assistant":
+            combined.append(f"### VOICE ({datetime.now().strftime('%Y-%m-%d %H:%M')}):\n{m['content']}")
+
+    # PDF
+    for m in SESSION_HISTORY.get(session_pdf, []):
+        if m["role"] == "assistant":
+            combined.append(f"### PDF ({datetime.now().strftime('%Y-%m-%d %H:%M')}):\n{m['content']}")
+
+    # Image
+    for m in SESSION_HISTORY.get(session_image, []):
+        if m["role"] == "assistant":
+            combined.append(f"### IMAGE ({datetime.now().strftime('%Y-%m-%d %H:%M')}):\n{m['content']}")
+
+    summary_text = "\n\n".join(combined)
+    if not summary_text.strip():
+        summary_text = "No summary available."
+
+    return generate_pdf_file(summary_text, "full_summary")
+
 # ------------------ Voice & Chat Handlers ------------------
 def _append_chat_display(session_id, user_text, assistant_text):
     if session_id not in CHAT_DISPLAY:
@@ -286,8 +310,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     <style>
         /* Change height + width of the audio recorder box */
         #mic_box audio {
-            height: 50px !important;   /* adjust height */
-            width: 200px !important;    /* adjust width (optional) */
+            height: 50px !important;
+            width: 200px !important;
         }
     </style>
     """)
@@ -296,8 +320,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     session_voice = gr.State(str(uuid.uuid4()))
     session_pdf = gr.State(str(uuid.uuid4()))
     session_image = gr.State(str(uuid.uuid4()))
-# FIX: define pdf_summary_file BEFORE it is used
-    #pdf_summary_file = gr.File(label="Download Summary", visible=False)
 
     with gr.Tab("🎤 Voice Chat"):
         chat_voice = gr.Chatbot(type="messages", height=300)
@@ -316,13 +338,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             btn_reset_logs = gr.Button("♻ Reset LOGs")
             btn_download_logs = gr.Button("📥 Download Summary")
             Voice_summary_file = gr.File(label="📥Download Summary File", interactive=False,scale =1)
-            #btn_general = gr.Button("⚡Ask General 🎯")
-            #btn_pdf = gr.Button("⚡Ask PDF 📄")
-            #btn_image = gr.Button("⚡Ask Image 🖼")
-        #with gr.Row():
-            #text_input = gr.Textbox(label="Or type a question (General)",visible=False)
-            #btn_send_text = gr.Button("Send (Text General)",visible=False)
-            #btn_reset_logs = gr.Button("♻ Reset LOGs")
         answer_voice = gr.Textbox(label="Assistant Answer (text)", lines=2, visible=False)
 
         btn_general.click(fn=handle_voice_general, 
@@ -330,9 +345,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                           outputs=[answer_voice, audio_output, chat_voice])
         btn_pdf.click(fn=handle_voice_pdf, inputs=[mic, session_pdf, tts_lang], outputs=[answer_voice, audio_output, chat_voice])
         btn_image.click(fn=handle_voice_image, inputs=[mic, session_image, tts_lang], outputs=[answer_voice, audio_output, chat_voice])
-       # btn_send_text.click(fn=handle_text_general, inputs=[text_input, session_voice, enhancer_toggle, tone_dropdown], outputs=[answer_voice, chat_voice])
         btn_reset_logs.click(lambda: (str(uuid.uuid4()), [], None, None, ""), outputs=[session_voice, chat_voice, mic, audio_output, answer_voice])
-        btn_download_logs.click(download_pdf_summary, inputs=[session_voice], outputs=[Voice_summary_file])
+
+        # <-- UPDATED: Full merged summary download
+        btn_download_logs.click(
+            download_full_summary,
+            inputs=[session_voice, session_pdf, session_image],
+            outputs=[Voice_summary_file]
+        )
         
     with gr.Tab("📄 PDF Summarizer"):
         pdf_output = gr.Textbox(label="Answer (Text Only)", lines=5)
@@ -369,8 +389,3 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         
         image_upload_btn.upload(handle_image_upload, inputs=[image_upload_btn, session_image], outputs=[image_upload_msg, image_output])
         image_send_btn.click(handle_text_image, inputs=[image_question, session_image], outputs=[image_output])
-        image_reset_btn.click(lambda: (str(uuid.uuid4()), ""), outputs=[session_image, image_output])
-        image_download_btn.click(download_pdf_summary, inputs=[session_image], outputs=[image_summary_file])
-
-if __name__ == "__main__":
-    demo.launch()