Update app.py

app.py

@@ -147,28 +147,31 @@ def user_query_with_rag(query, qa_chain, chatbot):
     if not query.strip():
         yield history, ""
         return
+
     if qa_chain is None:
-        history.append(("user", query))
-        history.append(("assistant", "Please upload and process a file first."))
+        history.append((query, "Please upload and process a file first."))
         yield history, ""
         return
 
-    history.append(("user", query))
-    history.append(("assistant", ""))
+    # Add user's question and placeholder for assistant's answer
+    history.append((query, ""))
     yield history, ""
 
     try:
         response = qa_chain.invoke({"question": query, "chat_history": []})
         assistant_response = response.get("answer", "No answer generated.")
+
+        # Typing effect
         for i in range(len(assistant_response)):
-            history[-1] = ("assistant", assistant_response[:i + 1])
+            history[-1] = (query, assistant_response[:i + 1])
             yield history, ""
             time.sleep(0.01)
     except Exception as e:
         logger.error(f"RAG error: {e}")
-        history[-1] = ("assistant", "An error occurred while answering your question.")
+        history[-1] = (query, "An error occurred while answering your question.")
         yield history, ""
 
+
 def demo():
     with gr.Blocks(title="RAG Analyzer") as app:
         db_state = gr.State(None)