Update app.py

app.py

@@ -276,7 +276,15 @@ def query_teapot(prompt, context, user_input, teapot_ai):
     return response
 
 @log_time
-def handle_chat(user_input, teapot_ai):
+def handle_chat(prompt, user_input, teapot_ai):
+    with st.chat_message("user"):
+        st.markdown(user_input)
+        st.session_state.messages.append({"role": "user", "content": user_input})
+        
+    with st.chat_message("assistant"):
+        st.markdown(response)
+        st.session_state.messages.append({"role": "assistant", "content": response})
+    
     results = brave_search(user_input)
     
     documents = [desc.replace('<strong>','').replace('</strong>','') for _, desc, _ in results]
@@ -299,28 +307,15 @@ def suggestion_button(suggestion_text, teapot_ai):
     if st.button(suggestion_text):
         handle_chat(suggestion_text, teapot_ai)
 
-@log_time
-def hash_documents(documents):
-    return hashlib.sha256("\n".join(documents).encode("utf-8")).hexdigest()
 
 def main():
     st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
     
     st.sidebar.header("Retrieval Augmented Generation")
-    user_documents = st.sidebar.text_area("Enter documents, each on a new line", value="\n".join(default_documents))
-    
-    documents = [doc.strip() for doc in user_documents.split("\n") if doc.strip()]
-    new_documents_hash = hash_documents(documents)
+    user_prompt = st.sidebar.text_area("Enter prompt, leave empty for search", value="\n".join(default_documents))
     
-    if "documents_hash" not in st.session_state or st.session_state.documents_hash != new_documents_hash:
-        with st.spinner('Loading Model and Embeddings...'):
-            start_time = time.time()
-            teapot_ai = TeapotAI(documents=documents or default_documents, settings=TeapotAISettings(rag_num_results=3,log_level="debug"))
-            end_time = time.time()
-            print(f"Model loaded in {end_time - start_time:.4f} seconds")
-        
-        st.session_state.documents_hash = new_documents_hash
-        st.session_state.teapot_ai = teapot_ai
+    teapot_ai = TeapotAI(documents=[], settings=TeapotAISettings(rag_num_results=3, log_level="debug"))
+
     else:
         teapot_ai = st.session_state.teapot_ai
     
@@ -335,25 +330,16 @@ def main():
     
     s1, s2, s3 = st.columns([1, 2, 3])
     with s1:
-        suggestion_button("Tell me about TeapotLLM.", teapot_ai)
+        suggestion_button("Tell me about teapotllm", teapot_ai)
     with s2:
-        suggestion_button("Who invented quantum mechanics", teapot_ai)
+        suggestion_button("Who invented quantum mechanics?", teapot_ai)
     with s3:
         suggestion_button("Extract the year Google was founded", teapot_ai)
     
     if user_input:
-        with st.chat_message("user"):
-            st.markdown(user_input)
-        
-        st.session_state.messages.append({"role": "user", "content": user_input})
         with st.spinner('Generating Response...'):
-            response = handle_chat(user_input, teapot_ai)
-        
-        with st.chat_message("assistant"):
-            st.markdown(response)
-        
-        st.session_state.messages.append({"role": "assistant", "content": response})
-        st.markdown("### Suggested Questions")
+            response = handle_chat(user_prompt, user_input, teapot_ai)
+       
 
 if __name__ == "__main__":
     main()