Update app.py

app.py

@@ -60,7 +60,7 @@ def brave_search(query, count=3):
 
 @traceable 
 @log_time
-def query_teapot(prompt, context, user_input, teapot_ai):
+def query_teapot(prompt, context, user_input):
     input_text = prompt + "\n" + context + "\n" + user_input
     
     start_time = time.time()
@@ -83,7 +83,7 @@ def query_teapot(prompt, context, user_input, teapot_ai):
 
 
 @log_time
-def handle_chat(user_prompt, user_input, teapot_ai):
+def handle_chat(user_prompt, user_input):
     with st.chat_message("user"):
         st.markdown(user_input)
         st.session_state.messages.append({"role": "user", "content": user_input})
@@ -102,7 +102,7 @@ def handle_chat(user_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. If a user asks who you are reply "I am Teapot"."""
-    response = query_teapot(prompt, context+user_prompt, user_input, teapot_ai)
+    response = query_teapot(prompt, context+user_prompt, user_input)
 
     with st.chat_message("assistant"):
         st.markdown(response)
@@ -156,7 +156,7 @@ def main():
         
     if user_input:
         with st.spinner('Generating Response...'):
-            response = handle_chat(user_prompt, user_suggested_input or user_input, teapot_ai)
+            response = handle_chat(user_prompt, user_suggested_input or user_input)
            
 
 if __name__ == "__main__":