Update app.py

app.py

@@ -2,11 +2,11 @@ import os
 import uuid
 import gradio as gr
 import openai
-import whisper
 import pdfplumber
 import docx
 import pandas as pd
 from PIL import Image
+import whisper
 
 # ====== UTILS SECTION ======
 def extract_text_from_file(file):
@@ -74,10 +74,10 @@ def process_message_with_files(message, file, audio, chat_history):
     full_message = "\n".join(filter(None, [message, file_text, audio_text]))
 
     if not full_message.strip():
-        return chat_history, "", None, None, "Please enter a message or upload a file/audio."
+        return chat_history, message, None, None, "Please enter a message or upload a file/audio."
 
     reply = generate_reply_with_context(chat_history, full_message)
-    chat_history.append(("user", message))
+    chat_history.append(("user", message.strip() or file_text or audio_text))
     chat_history.append(("assistant", reply))
     return chat_history, "", None, None, "Response generated."