Update app.py

app.py

@@ -83,6 +83,17 @@ def update_count_html():
     </div>
 """)
     return count_html
+    
+def store_message(db,query,answer,cross_encoder):
+    new_completion = db.collection('Nirvachana').document('chatlogs')
+    new_completion.set({
+        'query': query,
+        'answer':answer,
+        'created_time': firestore.SERVER_TIMESTAMP,
+        'embedding': cross_encoder,
+        'title': 'Expenditure observer bot'
+    })
+
 
 def add_text(history, text):
     history = [] if history is None else history
@@ -120,6 +131,7 @@ def bot(history, cross_encoder):
             history[-1][1] = character
             print('Final history is ',history)
             yield history, prompt_html
+        store_message(db,history[-1][0],history[-1][1],cross_encoder)
     else:
         # Retrieve documents relevant to query
         document_start = perf_counter()
@@ -162,6 +174,7 @@ def bot(history, cross_encoder):
             history[-1][1] = character
             print('Final history is ',history)
             yield history, prompt_html
+        store_message(db,history[-1][0],history[-1][1],cross_encoder)
 
 
 with gr.Blocks(theme='Insuz/SimpleIndigo') as demo: