Update app.py

app.py

@@ -276,7 +276,7 @@ def query_teapot(prompt, context, user_input, teapot_ai):
     return response
 
 @log_time
-def handle_chat(prompt, user_input, teapot_ai):
+def handle_chat(user_prompt, user_input, teapot_ai):
     with st.chat_message("user"):
         st.markdown(user_input)
         st.session_state.messages.append({"role": "user", "content": user_input})
@@ -295,7 +295,7 @@ def handle_chat(prompt, user_input, teapot_ai):
 
     context = "\n".join(documents)
     prompt = "You are Teapot, an open-source AI assistant optimized for low-end devices, providing short, accurate responses without hallucinating while excelling at information extraction and text summarization."
-    response = query_teapot(prompt, context, user_input, teapot_ai)
+    response = query_teapot(prompt, context+user_prompt, user_input, teapot_ai)
 
     with st.chat_message("assistant"):
         st.markdown(response)
@@ -312,7 +312,7 @@ def main():
     st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
     
     st.sidebar.header("Retrieval Augmented Generation")
-    user_prompt = st.sidebar.text_area("Enter prompt, leave empty for search", value="\n".join(default_documents))
+    user_prompt = st.sidebar.text_area("Enter prompt, leave empty for search")
     
     teapot_ai = TeapotAI(documents=[], settings=TeapotAISettings(rag_num_results=3, log_level="debug"))