Update app.py

app.py

@@ -262,10 +262,11 @@ def process_question(question, history):
         return "Please upload a document first.", history
     try:
         response = rag.ask_question(question)
-        history.append((question, response))
+        history.append({"role": "user", "content": question})
+        history.append({"role": "assistant", "content": response})
         return "", history
     except Exception as e:
-        return "", history + [(question, f"Error: {str(e)}")]
+        return "", history + [{"role": "assistant", "content": f"Error: {str(e)}"}]
 
 def process_audio(audio_file, history):
     if audio_file is None:
@@ -275,10 +276,11 @@ def process_audio(audio_file, history):
         if not rag.thread_id:
             return "Please upload a document first.", history
         response = rag.ask_question(transcript)
-        history.append((transcript, response))
+        history.append({"role": "user", "content": f"🎤 {transcript}"})
+        history.append({"role": "assistant", "content": response})
         return "", history
     except Exception as e:
-        return "", history + [("Audio input", f"Error: {str(e)}")]
+        return "", history + [{"role": "assistant", "content": f"Error: {str(e)}"}]
 
 def process_audio_base64(audio_base64, history):
     if not audio_base64:
@@ -323,7 +325,7 @@ with gr.Blocks(title="Document Q&A System") as demo:
 
     with gr.Row():
         with gr.Column():
-            chatbot = gr.Chatbot(height=400)
+            chatbot = gr.Chatbot(height=400, type="messages")
             question = gr.Textbox(label="Ask a question")
             question.submit(process_question, [question, chatbot], [question, chatbot])
 
@@ -333,8 +335,12 @@ with gr.Blocks(title="Document Q&A System") as demo:
             audio_input = gr.Audio(label="Voice Input", type="filepath", visible=False)
             audio_input.change(process_audio, [audio_input, chatbot], [audio_input, chatbot])
 
-    # Handle audio data from custom recorder
-    demo.load(None, None, None, _js="""
+    # Handle audio data from custom recorder using JavaScript
+    demo.load(
+        fn=None,
+        inputs=None,
+        outputs=None,
+        _js="""
         function() {
             window.addEventListener('message', function(event) {
                 if (event.data.type === 'audio_data') {
@@ -359,7 +365,8 @@ with gr.Blocks(title="Document Q&A System") as demo:
                 }
             });
         }
-    """)
+        """
+    )
 
 if __name__ == "__main__":
     demo.launch(