Update app.py

app.py

@@ -1,55 +1,30 @@
-
 import streamlit as st
+from transformers import Conversation, pipeline
 
+st.set_page_config(page_title="Dolphin Chatbot", page_icon=":robot_face:")
+st.header("Dolphin 2.6 Mistral 7B Chatbot")
 
-from langchain.chat_models import ChatOpenAI
-from langchain.schema import (
-    AIMessage,
-    HumanMessage,
-    SystemMessage
-)
-
-# From here down is all the StreamLit UI
-st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
-st.header("IPE Chatbot")
-
-
-
-if "sessionMessages" not in st.session_state:
-     st.session_state.sessionMessages = [
-        SystemMessage(content="You are customer support chatbot from a website.")
-    ]
-
+# Initialize the conversation pipeline with dolphin-2.6-mistral-7b model
+conversational_pipeline = pipeline('conversational', model='google/flan-t5-large')
 
+if "conversation" not in st.session_state:
+    st.session_state.conversation = Conversation()
 
 def load_answer(question):
-
-    st.session_state.sessionMessages.append(HumanMessage(content=question))
-
-    assistant_answer  = chat(st.session_state.sessionMessages )
-
-    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
-
-    return assistant_answer.content
-
+    new_user_input = Conversation(text=question)
+    st.session_state.conversation.add_user_input(question)
+    assistant_answer = conversational_pipeline(st.session_state.conversation)
+    st.session_state.conversation = assistant_answer
+    return assistant_answer.text
 
 def get_text():
-    input_text = st.text_input("You: ", key= input)
+    input_text = st.text_input("You: ", key="input_text")
     return input_text
 
-
-chat = ChatOpenAI(temperature=0)
-
-
-
-
-user_input=get_text()
-submit = st.button('Generate')  
+user_input = get_text()
+submit = st.button('Generate')
 
 if submit:
-    
     response = load_answer(user_input)
     st.subheader("Answer:")
-
-    st.write(response,key= 1)
-
+    st.write(response)