Update app.py

app.py

@@ -11,7 +11,7 @@ from pathlib import Path
 from time import sleep
 import random
 import string
-
+import sys
 import os
 from dotenv import load_dotenv
 load_dotenv()
@@ -19,9 +19,9 @@ load_dotenv()
 st.set_page_config(page_title="Open AI Doc-Chat Assistant", layout="wide")
 st.subheader("Open AI Doc-Chat Assistant: Life Enhancing with AI!")
 
-#css_file = "main.css"
-#with open(css_file) as f:
-#    st.markdown("<style>{}</style>".format(f.read()), unsafe_allow_html=True)
+css_file = "main.css"
+with open(css_file) as f:
+    st.markdown("<style>{}</style>".format(f.read()), unsafe_allow_html=True)
     
 HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
@@ -45,36 +45,32 @@ new_index = VectorStoreIndex.from_documents(
 )
 
 new_index.storage_context.persist("random_string")
-
 storage_context = StorageContext.from_defaults(persist_dir="random_string")
-
 loadedindex = load_index_from_storage(storage_context=storage_context, service_context=service_context)
-
 query_engine = loadedindex.as_query_engine()
 
-while True:
-    try:
-        question = st.text_input("Enter your query here:")
-        print("Your query:\n"+question)
-        if question.strip().isspace():
-            st.write("Query Empty. Please enter valid query first.")
-            break
-        elif question == "":
-#            st.write("Query Empty. Please enter valid query first.")
-            break
-        elif question.strip() == "":
-            st.write("Query Empty. Please enter valid query first.")
-            break
-        elif question.isspace():
-            st.write("Query Empty. Please enter valid query first.")
-            break
-        elif question=="exit":
-            break
-        elif question!="":
-            initial_response = query_engine.query(question)
-            temp_ai_response=str(initial_response)
-            final_ai_response=temp_ai_response.partition('<|end|>')[0] 
-            print("AI Response:\n"+final_ai_response)
-            st.write("AI Response:\n\n"+final_ai_response)
-    except Exception as e:
-        st.stop()
+question = st.text_input("Enter your query here:")
+display_output_text = st.checkbox("Check AI Repsonse", key="key_checkbox", help="Check me to get AI Response.")
+
+with st.sidebar:    
+    st.subheader("Valuation.pdf furnished background!")
+    st.write("Disclaimer: This app is for information purpose only. NO liability could be claimed against whoever associated with this app in any manner. User should consult a qualified legal professional for legal advice.")
+    st.sidebar.markdown("Contact: [[email protected]](mailto:[email protected])")
+    st.sidebar.markdown('WeChat: <span class="blue-underline">pat2win</span>, or scan the code below.', unsafe_allow_html=True)
+    st.image(wechat_image)
+    st.sidebar.markdown('<span class="blue-underline">Life Enhancing with AI.</span>', unsafe_allow_html=True)      
+    st.subheader("Enjoy chatting!")
+
+if question !="" and not question.strip().isspace() and not question == "" and not question.strip() == "" and not question.isspace():
+    if display_output_text==True:
+      with st.spinner("AI Thinking...Please wait a while to Cheers!"):
+        initial_response = query_engine.query(question)
+        temp_ai_response=str(initial_response)
+        final_ai_response=temp_ai_response.partition('<|end|>')[0]
+        st.write("AI Response:\n\n"+final_ai_response)
+    else:
+        print("Check the Checkbox to get AI Response.")
+        sys.exit()          
+else:
+    print("Please enter your question first.")
+    st.stop()