Update app.py

app.py

@@ -115,6 +115,26 @@ def load_llm(huggingface_repo_id, hf_token):
         model_kwargs={"max_length": 512}
     )
 
+def handle_translation():
+    if "last_response" in st.session_state and st.session_state.last_response:
+        try:
+            if not st.session_state.get("translation_done", False):
+                translated_text = translate_text(st.session_state.last_response, "hi")
+
+                # Update the last assistant message
+                for i in range(len(st.session_state.messages) - 1, -1, -1):
+                    if st.session_state.messages[i]["role"] == "assistant":
+                        st.session_state.messages[i]["content"] = f'<div class="hindi-text">{translated_text}</div>'
+                        break
+
+                # Mark translation as done
+                st.session_state.translation_done = True
+                st.rerun()  # Forces a UI refresh
+
+        except Exception as e:
+            st.error(f"Translation error: {str(e)}")
+    
+
 def format_source_docs(source_documents):
     formatted_docs = []
     for idx, doc in enumerate(source_documents, start=1):
@@ -174,23 +194,23 @@ def handle_user_input(prompt, qa_chain):
             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:
-        try:
-            if not st.session_state.get("translation_done", False):
-                translated_text = translate_text(st.session_state.last_response, "hi")
+# def handle_translation():
+#     if "last_response" in st.session_state and st.session_state.last_response:
+#         try:
+#             if not st.session_state.get("translation_done", False):
+#                 translated_text = translate_text(st.session_state.last_response, "hi")
                 
-                # Update messages
-                for msg in reversed(st.session_state.messages):
-                    if msg["role"] == "assistant":
-                        msg["content"] = f'<div class="hindi-text">{translated_text}</div>'
-                        break
+#                 # Update messages
+#                 for msg in reversed(st.session_state.messages):
+#                     if msg["role"] == "assistant":
+#                         msg["content"] = f'<div class="hindi-text">{translated_text}</div>'
+#                         break
                 
-                st.session_state.translation_done = True
-                st.rerun()  # Corrected rerun method
+#                 st.session_state.translation_done = True
+#                 st.rerun()  # Corrected rerun method
                 
-        except Exception as e:
-            st.error(f"Translation error: {str(e)}")
+#         except Exception as e:
+#             st.error(f"Translation error: {str(e)}")
 
 def render_chat_messages():
     for message in st.session_state.messages: