Update app.py

app.py

@@ -40,29 +40,7 @@ def Bot(Questions):
     llama3 = Together(model="meta-llama/Llama-3-70b-chat-hf", max_tokens=250)
     Generated_chat = LLMChain(llm=llama3, prompt=prompt)
 
-    try:
-        response = Generated_chat.invoke({
-            "text": text,
-            "Questions": Questions
-        })
-
-        response_text = response['text']
-
-        response_text = response_text.replace("assistant", "")
-
-        # Post-processing to handle repeated words and ensure completeness
-        words = response_text.split()
-        seen = set()
-        filtered_words = [word for word in words if word.lower() not in seen and not seen.add(word.lower())]
-        response_text = ' '.join(filtered_words)
-        response_text = response_text.strip()  # Ensuring no extra spaces at the ends
-        if not response_text.endswith('.'):
-            response_text += '.'
-
-        return response_text
-    except Exception as e:
-        return f"Error in generating response: {e}"
-
+   
 def ChatBot(Questions):
   greetings = ["hi", "hello", "hey", "greetings", "what's up", "howdy"]
     # Check if the input question is a greeting
@@ -106,4 +84,25 @@ if prompt := st.chat_input():
 
     with st.chat_message("assistant"):
         message_placeholder = st.empty()
-        full_response = ""
+        full_response = ""
+
+        try:
+            _chat_history = st.session_state.langchain_messages[1:40]
+            _chat_history_tranform = list(
+                chunked([msg.content for msg in _chat_history], n=2)
+            )
+
+            response = rag_chain.stream(
+                {"question": prompt, "chat_history": _chat_history_tranform}
+            )
+
+            for res in response:
+                full_response += res or ""
+                message_placeholder.markdown(full_response + "|")
+                message_placeholder.markdown(full_response)
+
+            msgs.add_user_message(prompt)
+            msgs.add_ai_message(full_response)
+
+        except Exception as e:
+            st.error(f"An error occured. {e}")