Update app.py

app.py

@@ -6,6 +6,15 @@ import streamlit as st
 from PIL import Image
 from functools import partial
 
+def generate_response(question):
+    response = st.session_state.vq.submit_query(question)
+    return response
+
+def generate_streaming_response(question):
+    response = st.session_state.vq.submit_query_streaming(question)
+    return response
+
+
 def set_query(q: str):
     st.session_state.messages.append({"role": "user", "content": q})
     if st.session_state.cfg.streaming:
@@ -19,14 +28,8 @@ def set_query(q: str):
     st.session_state.messages.append(message)
 
 
+
 def launch_bot():
-    def generate_response(question):
-        response = vq.submit_query(question)
-        return response
-    
-    def generate_streaming_response(question):
-        response = vq.submit_query_streaming(question)
-        return response
 
     if 'cfg' not in st.session_state:
         corpus_ids = str(os.environ['corpus_ids']).split(',')