cgpt check 2

app.py

@@ -1,13 +1,12 @@
 import streamlit as st
 import os
-from langchain_community.vectorstores import Chroma
-from langchain_community.embeddings import HuggingFaceBgeEmbeddings
-from langchain_community.llms import Together
+from langchain.vectorstores import Chroma
+from langchain.embeddings import HuggingFaceBgeEmbeddings
+from langchain.llms import Together
 from langchain.prompts import ChatPromptTemplate, PromptTemplate
 from langchain.schema import format_document
 from typing import List
 from langchain.memory import ConversationBufferMemory
-from langchain.schema.runnable import RunnableParallel, RunnablePassthrough, StrOutputParser
 from langchain_core.chat_message_histories import StreamlitChatMessageHistory
 import time
 
@@ -71,7 +70,7 @@ def get_streaming_response(user_query, chat_history):
         "user_question": user_query
     }
 
-    chain = prompt | llm | StrOutputParser()
+    chain = prompt | llm
     return chain.stream(inputs)
 
 def app():
@@ -117,12 +116,15 @@ def app():
         with st.chat_message("assistant"):
             with st.spinner("Thinking..."):
                 chat_history = "\n".join([f"{msg['role']}: {msg['content']}" for msg in chistory])
-                response_generator = get_streaming_response(user_query, chat_history)
-                response_text = ""
-                for response_part in response_generator:
-                    response_text += response_part
-                    st.write(response_text)
-                st.session_state.messages.append({"role": "assistant", "content": response_text})
+                try:
+                    response_generator = get_streaming_response(user_query, chat_history)
+                    response_text = ""
+                    for response_part in response_generator:
+                        response_text += response_part
+                        st.write(response_text)
+                    st.session_state.messages.append({"role": "assistant", "content": response_text})
+                except Exception as e:
+                    st.error(f"An error occurred: {e}")
 
 if __name__ == '__main__':
     app()