Update app.py

app.py

@@ -1,12 +1,12 @@
 import streamlit as st
 import random
 import time
-import app_server
-
+from app_server import res
 # Streamed response emulator
 def response_generator(p1):
     print(p1)
-    response = str(app_server.res(p1))
+    response = str(res(p1))
+    #response = str(st.session_state.res(p1))
     # return response
     for word in response.split("\n"):
         yield word + " " + "\n"
@@ -35,4 +35,5 @@ if prompt := st.chat_input("What is up?"):
     with st.chat_message("assistant"):
         response = st.write_stream(response_generator(prompt))
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
+    st.session_state.messages.append({"role": "assistant", "content": response})
+