Update app.py

app.py

@@ -26,7 +26,7 @@ Don't Try To Make Up An Answer. Don't Provide Anything Out Of The Given Context.
 Context: {context}
 Question: {question}
 
-Start The Answer Directly. No Small Talk, Please. The Answer Should Contain All 3 Contexts.
+Start The Answer Directly., Please. The Answer Should Contain All 3 Contexts.
 Consider Yourself As God Krishna And Answer The Question Result Should Not Start With "Answer"
 """  # Keep your template here
 
@@ -37,7 +37,8 @@ def initialize_session_states():
         "selected_question": None,
         "show_predefined": True,
         "last_response": None,
-        "translation_done": False
+        "translation_done": False,
+        "last_prompt": None  # Add this line
     }
     for key, val in session_defaults.items():
         if key not in st.session_state:
@@ -124,33 +125,54 @@ def format_source_docs(source_documents):
 
 def handle_user_input(prompt, qa_chain):
     if prompt:
+        # Check if this prompt has already been processed
+        if st.session_state.get("last_prompt") == prompt:
+            return
+        
+        # Store the current prompt to prevent reprocessing
+        st.session_state.last_prompt = prompt
+        
         with st.chat_message("user", avatar="🐿"):
             st.markdown(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
 
         try:
+            # Add temporary assistant message
+            with st.chat_message("assistant", avatar="🪈"):
+                response_placeholder = st.empty()
+            
+            # Process query and generate response
             response = qa_chain.invoke({"query": prompt})
             result = response["result"]
             source_documents = response["source_documents"]
 
-            response_placeholder = st.empty()
+            # Build response incrementally
             accumulated_text = ""
             for char in result:
                 accumulated_text += char
                 response_placeholder.markdown(f'<div class="english-text">{accumulated_text}</div>', unsafe_allow_html=True)
                 time.sleep(0.01)
             
-            if "don't have information" not in result.lower():
-                with st.expander("Source Documents"):
-                    st.markdown(format_source_docs(source_documents))
-
-            st.session_state.messages.append({"role": "assistant", "content": accumulated_text})
+            # Update session state with final response
+            st.session_state.messages.append({
+                "role": "assistant",
+                "content": f'<div class="english-text">{accumulated_text}</div>',
+                "original": accumulated_text
+            })
+            
             st.session_state.last_response = accumulated_text
             st.session_state.show_predefined = False
             st.session_state.translation_done = False
 
+            if "don't have information" not in result.lower():
+                with st.expander("Source Documents"):
+                    st.markdown(format_source_docs(source_documents))
+
         except Exception as e:
             st.error(f"Error: {str(e)}")
+            # Remove temporary assistant message on error
+            if st.session_state.messages and st.session_state.messages[-1]["role"] == "assistant":
+                st.session_state.messages.pop()
 
 def handle_translation():
     if "last_response" in st.session_state and st.session_state.last_response: