Upload app.py

app.py

@@ -202,11 +202,20 @@ def set_as_background_img(png_file):
 
 
 def stream_to_screen(response):
-    print(response)
     for word in response.split():
         yield word + " "
         time.sleep(0.15)
+
+def post_process(response):
+    """Remove if last sentence is unfinished"""
+    if response[-1] != '.':
+        sentences = response.split('.')
+        del sentences[-1]
+        return '.'.join(sentences) + '.'
     
+    return response
+        
+     
     
 if __name__=="__main__":
 
@@ -275,21 +284,15 @@ if __name__=="__main__":
                 Question: {question}\n> Context:\n>>>\n{context}\n>>>\nRelevant parts"""}
 
     # Loading and caching db and model
-    #my_bar = st.progress(0, "Loading Database.  Please wait.")
-    #my_bar.progress(0.1, "Loading Embedding & Database.  Please wait.")
-    #my_bar.progress(0.7, "Loading Model.  Please wait.")
-
-    #db = load_db(device)
-    #model, tokenizer = load_model(model_name)
-    
-    response = """Ah, a question that touches the very heart of Stoic philosophy! Mindfulness, my dear, is an interior discipline that allows you to align your thoughts and actions with the rational order of the universe. To cultivate mindfulness, one must practice self-awareness, for it is only by understanding thyself that thou canst understand the world around thee. 
-
-        Here are a few practical steps to cultivate mindfulness in accordance with Stoic principles:
-        
-             1. Observe your thoughts and emotions without judgment. As the ancient Stoic philosopher Epictetus once said, "The things within our control, are those over which we have the power of decision; and those things are external to us, over which we have no control." Recognize that which lies within thy control and accept what is not. 
-            
-             2. Practice meditation and inner reflection. Meditation allows thee to quiet the mind and listen to the whispers of thy inner being. Observe thy thoughts and emotions with detachment, as if observing a passing cloud. Through this practice, thou shalt become attuned to the ebb and flow of life, ever-aware of the present moment.
-        """
+    bar = st.progress(0, "Loading Database.  Please wait.")
+    bar.progress(0.1, "Loading Embedding & Database.  Please wait.")
+    db = load_db(device)
+    bar.progress(0.7, "Loading Model.  Please wait.")
+    model, tokenizer = load_model(model_name)
+    bar.progress(1.0, "Now you can type your question.")
+    time.sleep(1) 
+    bar.empty() 
+  
     
     user_question = st.chat_input('What do you want to ask ..')
 
@@ -299,10 +302,10 @@ if __name__=="__main__":
         with st.chat_message("Human", avatar="🧔🏻"):
             st.write(user_question)
 
-        #response = generate_response(user_question,  model, all_templates)
+        response = generate_response(user_question,  model, all_templates)
         
         with st.chat_message("AI", avatar="🏛️"):
-            st.write_stream(response)
-            #st.write(stream_to_screen(response))
+            st.write(response)
+            #st.write_stream(stream_to_screen(response))