Update app.py

app.py

@@ -171,13 +171,28 @@ def generate_response(query,  model, template):
     logger.info(f"Total Execution Time: {time.time() - start_time}") 
 
     my_bar.progress(0.9, "Post Processing.  Please wait.")
-    
+    response = post_process(response)
+
     my_bar.progress(1.0, "Done")
     time.sleep(1) 
     my_bar.empty()  
     return response
 
 
+def stream_to_screen(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
+    
 # show background image
 def convert_to_base64(bin_file):
     with open(bin_file, 'rb') as f:
@@ -201,21 +216,7 @@ def set_as_background_img(png_file):
     return   
 
 
-def stream_to_screen(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__":
 
@@ -283,6 +284,7 @@ if __name__=="__main__":
 
                 Question: {question}\n> Context:\n>>>\n{context}\n>>>\nRelevant parts"""}
 
+    
     # Loading and caching db and model
     bar = st.progress(0, "Loading Database.  Please wait.")
     bar.progress(0.1, "Loading Embedding & Database.  Please wait.")
@@ -303,7 +305,7 @@ if __name__=="__main__":
             st.write(user_question)
 
         response = generate_response(user_question,  model, all_templates)
-        
+
         with st.chat_message("AI", avatar="🏛️"):
             st.write(response)
             #st.write_stream(stream_to_screen(response))