Update app.py

app.py

@@ -1,10 +1,9 @@
 
-"""
-Credit to Derek Thomas, [email protected]
-"""
 from ragatouille import RAGPretrainedModel
 import subprocess
-
+import json
+import firebase_admin
+from firebase_admin import credentials, firestore
 import logging
 from pathlib import Path
 from time import perf_counter
@@ -31,15 +30,58 @@ env = Environment(loader=FileSystemLoader(proj_dir / 'templates'))
 # Load the templates directly from the environment
 template = env.get_template('template.j2')
 template_html = env.get_template('template_html.j2')
+service_account_key='firebase.json'
+# Create a Certificate object from the service account info
+cred = credentials.Certificate(service_account_key)
+# Initialize the Firebase Admin SDK
+firebase_admin.initialize_app(cred)
+
+# # Create a reference to the Firestore database
+db = firestore.client()
 
-# crossEncoder
-#cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2') 
-#cross_encoder = CrossEncoder('BAAI/bge-reranker-base')
 # Examples
 examples = ['what is social media and what are rules related to it for expenditure monitoring ',
             'how many reports to be submitted by Expenditure observer with annexure names ?','what is expenditure limits for parlimentary constituency and assembly constituency'
             ]
-
+#db usage
+collection_name = 'Nirvachana'  # Replace with your collection name
+field_name = 'message_count'  # Replace with your field name for count
+
+
+def get_and_increment_value_count(db , collection_name, field_name):
+    """
+    Retrieves a value count from the specified Firestore collection and field,
+    increments it by 1, and updates the field with the new value."""
+    collection_ref = db.collection(collection_name)
+    doc_ref = collection_ref.document('count_doc')  # Assuming a dedicated document for count
+
+    # Use a transaction to ensure consistency across reads and writes
+    try:
+        with db.transaction() as transaction:
+            # Get the current value count (or initialize to 0 if it doesn't exist)
+            current_count_doc = doc_ref.get()
+            current_count_data = current_count_doc.to_dict()
+            if current_count_data:
+                current_count = current_count_data.get(field_name, 0)
+            else:
+                current_count = 0
+            # Increment the count
+            new_count = current_count + 1
+            # Update the document with the new count
+            transaction.set(doc_ref, {field_name: new_count})
+            return new_count
+    except Exception as e:
+        print(f"Error retrieving and updating value count: {e}")
+        return None  # Indicate error
+        
+def update_count_html():
+    usage_count = get_and_increment_value_count(db ,collection_name, field_name)
+    count_html =gr.HTML(value= f"""
+    <div style="display: flex; justify-content: flex-end;">
+    <span style="font-weight: bold; color: maroon;">No of Usages:</span> {usage_count}
+    </div>
+    """)
+    return count_html
 
 def add_text(history, text):
     history = [] if history is None else history
@@ -127,8 +169,13 @@ with gr.Blocks(theme='Insuz/SimpleIndigo') as demo:
     <img src='logo.png' alt="Chatbot" width="50" height="50" />
     </div>""",elem_id='heading')
     gr.HTML(value="""<p style="font-family: sans-serif; font-size: 16px;">A free chat bot assistant for Expenditure Observers on Compendium on Election Expenditure Monitoring using Open source LLMs. <br> The bot can answer questions in natural language, taking relevant extracts from the ECI document which can be accessed <a href="https://www.eci.gov.in/eci-backend/public/api/download?url=LMAhAK6sOPBp%2FNFF0iRfXbEB1EVSLT41NNLRjYNJJP1KivrUxbfqkDatmHy12e%2Fzk1vx4ptJpQsKYHA87guoLjnPUWtHeZgKtEqs%2FyzfTTYIC0newOHHOjl1rl0u3mJBSIq%2Fi7zDsrcP74v%2FKr8UNw%3D%3D" style="color: #FF0000; text-decoration: none;"></a>.</p>""",elem_id='Sub-heading')
+    usage_count = get_and_increment_value_count(collection_name, field_name)
+    count_html =gr.HTML(value= f"""
+    <div style="display: flex; justify-content: flex-end;">
+    <span style="font-weight: bold; color: maroon;">No of Usages:</span> {usage_count}
+    </div>
+    """)
     
-
     chatbot = gr.Chatbot(
             [],
             elem_id="chatbot",
@@ -153,14 +200,14 @@ with gr.Blocks(theme='Insuz/SimpleIndigo') as demo:
     prompt_html = gr.HTML()
     # Turn off interactivity while generating if you click
     txt_msg = txt_btn.click(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
-            bot, [chatbot, cross_encoder], [chatbot, prompt_html])
+            bot, [chatbot, cross_encoder], [chatbot, prompt_html]).then(update_count_html,[],[count_html])
 
     # Turn it back on
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
 
     # Turn off interactivity while generating if you hit enter
     txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
-            bot, [chatbot, cross_encoder], [chatbot, prompt_html])
+            bot, [chatbot, cross_encoder], [chatbot, prompt_html]).then(update_count_html,[],[count_html])
 
     # Turn it back on
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)