Update app.py

app.py

@@ -28,10 +28,36 @@ api = st.sidebar.text_input("API-key", type = "password")
 MODEL = st.sidebar.selectbox(label = "Model", options = ["gpt-3.5-turbo", "text-davinci-003"])
 
 if api:
+    
     llm = OpenAI(
         temperature = 0,
         open_api_key = api,
         model_name = MODEL
     )
+
+    #Create Conv Memory
+
+    if "entity_memory" not in st.session_state:
+        st.session_state.entity_memory = ConversationEntityMemory(
+            llm = llm,
+            k = 10
+        )
+
+    # Create Conv Chain
+
+    Conversation = ConversationChain(
+        llm = llm,
+        prompt = ENTITY_MEMORY_CONVERSATION_TEMPLATE,
+        memory = st.session_state.entity_memory
+    )
+
+else:
+    st.error("No API found!")
+
+
+if user_input:
+    output = Conversation.run(input = user_input)
+    st.session_state.past.append(user_input)
+    st.session_state.generated.append(output)