firestore added

app.py

@@ -1,6 +1,41 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import re
+import random
+import string
+import firebase_admin
+from firebase_admin import credentials, firestore
+from datetime import datetime
+
+# Initialiser Firebase
+cred = credentials.Certificate("./montaillou-bdd-5b8a59219187.json")
+firebase_admin.initialize_app(cred)
+
+# Obtenir une référence à la base de données Firestore
+db = firestore.client()
+
+
+def save_message(user_id, message):
+    # Enregistrer le message dans Firestore
+    doc_ref = db.collection("conversations").document(user_id)
+    doc_ref.set(
+        {
+            "messages": firestore.ArrayUnion(
+                [{"message": message, "timestamp": datetime.now()}]
+            )
+        }
+    )
+
+
+def load_history(user_id):
+    # Charger l'historique de conversation de l'utilisateur
+    doc_ref = db.collection("conversations").document(user_id)
+    doc = doc_ref.get()
+    if doc.exists:
+        messages = doc.to_dict()["messages"]
+        return messages
+    else:
+        return []
 
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -8,7 +43,6 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 # Name of the chat
 chat_name = "Tavernier 🍺"
 
-
 # Variable globale pour compter les itérations
 global_iteration = 0
 
@@ -22,7 +56,6 @@ global_context = {
     ],
 }
 
-
 # Créer une interface avec un champ de texte pour le contexte caché
 context_input = gr.Textbox(
     label="Contexte caché", value=global_context["static_context"]
@@ -63,8 +96,11 @@ def format_prompt(message, history, user_id=""):
     for user_prompt, bot_response in history:
         prompt += f"[USER] {user_prompt} [/USER]"
         if (
-            user_prompt == history[-1][0]
-        ):  # N'ajouter les balises [BOT] qu'autour de la dernière réponse du bot
+            history
+            and isinstance(history[-1], dict)
+            and user_prompt == history[-1].get("0")
+        ):
+            # N'ajouter les balises [BOT] qu'autour de la dernière réponse du bot
             prompt += f" [BOT]{filter_sensitive_info(bot_response)}[/BOT] "
         else:
             prompt += f" {filter_sensitive_info(bot_response)} "
@@ -76,6 +112,12 @@ def format_prompt(message, history, user_id=""):
     return prompt
 
 
+def get_random_string(length):
+    # choose from all lowercase letter
+    letters = string.ascii_lowercase
+    return "".join(random.choice(letters) for i in range(length))
+
+
 def generate(
     prompt,
     history,
@@ -100,8 +142,9 @@ def generate(
     )
 
     if request:
-        # Récupérer l'identifiant unique de l'utilisateur à partir de la requête HTML
+        # Récupérer l'identifiant unique de l'utilisateur à partir de la requête HTML et son historique de conversation
         user_id = dict(request.query_params).get("user_id")
+        history = load_history(user_id)
     else:
         user_id = ""
 
@@ -120,12 +163,20 @@ def generate(
         return_full_text=False,
     )
     output = ""
+    bot_response = ""
 
     for response in stream:
         response_text = filter_sensitive_info(response.token.text)
         response_text = remove_length_info(response_text)
         output += response_text
+        bot_response += response_text
         yield output
+
+    if user_id != "":
+        # Sauvegarder le message de l'utilisateur et la réponse complète du bot dans Firestore
+        save_message(user_id, prompt)
+        save_message(user_id, bot_response)
+
     return output
 
 
@@ -143,6 +194,13 @@ def chatbot_interface(request: gr.Request):
         title=chat_name,
         retry_btn=None,
         undo_btn=None,
+        examples=[
+            [
+                "Bonjour",
+                "Salutations, voyageur ! Bienvenue dans ma taverne. Que puis-je faire pour toi ?",
+            ]
+        ],
+        cache_examples=True,
     )
     return chatbot_interface
 
@@ -156,9 +214,6 @@ if __name__ == "__main__":
     main(None)
 
 
-# Créer une interface pour le chatbot avec un champ de texte pour le contexte caché
-
-
 # def reset_history(chatbot):
 #     chatbot.history = []
 #     return []
@@ -172,41 +227,3 @@ if __name__ == "__main__":
 #     global global_iteration
 #     global_iteration = 0
 #     return None
-
-
-# chatbot_interface = gr.ChatInterface(
-#     fn=generate,
-#     chatbot=mychatbot,
-#     title=chat_name,
-#     retry_btn=None,
-#     undo_btn=None,
-# )
-
-
-# Ajouter le champ de texte pour le contexte caché à l'interface
-# context_interface = gr.Interface(
-#     fn=update_context,
-#     inputs=[context_input],
-#     outputs=None,
-#     title="Modifier le contexte caché",
-#     description="Modifie le contexte caché du tavernier",
-#     article=False,
-# )
-
-# demo = gr.TabbedInterface([chatbot_interface, context_interface], title="Tavernier 🍺")
-
-# Lancer et afficher l'interface
-
-
-# import gradio as gr
-
-
-# def echo(text, request: gr.Request):
-#     if request:
-#         print("Request headers dictionary:", request.headers)
-#         print("IP address:", request.client.host)
-#         print("Query parameters:", dict(request.query_params))
-#     return text
-
-
-# io = gr.Interface(echo, "textbox", "textbox").launch()

gradio_cached_examples/14/log.csv

@@ -0,0 +1,2 @@
+component 0,flag,username,timestamp
+"[[""Bonjour"", ""Bonjour! Bienvenue dans ma taverne. Assois-toi et repose-toi. Tu veux peut-\u00eatre boire un coup? Nous avons du vin et de la bi\u00e8re. Je ne suis pas un noble, mais je suis fier de ce que je suis: un tavernier qui sert les gens honn\u00eates. Les cathares sont les bienvenus ici. Mais si tu es catholique ou un noble, je ne peux pas te promettre que tu te sentiras \u00e0 l'aise.</s>""]]",,,2024-03-26 11:07:18.013347

montaillou-bdd-5b8a59219187.json

@@ -0,0 +1,13 @@
+{
+  "type": "service_account",
+  "project_id": "montaillou-bdd",
+  "private_key_id": "5b8a59219187f785c82f11070f0c0ab1f2fea43a",
+  "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQDluiD1QdCDFrIF\nCLE/2edjw+f+ZdUWW2QoBpSEqdZB3cIxqmC/+Dt4TbKAvgBoiWuteZoF6jrjhu89\nNNYlSuhZ7ZN6hdiN62GF/2NkZfCGcH+L/AzJ/E8Z7Wu1AAmiEZS7px/BGnkdnF/e\n7EFYxoP10rXws+xdHlmGoGkd85CJIGkA2OxNk7UBADRaxIdSivYB/oJ8aRjebhx2\nTIg+Qn3HS8qQFg4Ni5sAGF2tznfqVE2lSg0GZ/wtMNckJgADWVCcbIYWldSXZ2H6\nSVvQtqPGC8uBllWdKPtGc49+dDTKOP+YDaPoxQ/P0Ci4W58zqW7hiLgHjs8DeXY+\nRKbFrwFNAgMBAAECggEACH6/DELASrCyqZP+Nf3BXEOY1yP94sNDa2V+5X8AUNQ6\nA6sow8CStmqtIKymCDeYt0QJaZGVVuec/jbvelaQTA8Ihp3febQMGa/thX9tixMq\n7ErIUEnXxP4uHvXe5VLgHKZb+sCXXPx04SH8cqN4qsT9fGay5WijZ+KTsItau29Y\n7LtP3FsB+pugvVw7euBnLK+/FrDoEBk4dZiAsp2Uf586OZvES5g+Ib8VSKiVTwYQ\naktu4Fb7uea1Iodi2O+b3FcF6gP/HzuT/dZN80sfZed5BdvaTMj97EXfXmWB3DQ7\nQNnINcGQDuKC+WOD3STn6+ZSPrtxkMhWqivmL4ZhMwKBgQD/rHX2eGmM8H2tdfQw\nPw/+R5h4GKtXwbK3aPcp/QXejV9aB1jIvmssFSk87sj15Kwa+CZaPY+C1oiC3ASC\nPZ5zuVVBQNUrL38JXSRrtI8mt7z9nAdEwjazUyEdBtRjX05GwiC8f8rMAcOGA/rl\nCBYJupXbrvrrMriAQ7tz/nf6wwKBgQDmBTCrZaF8WM3OWvdw5EZpAmU4IpELh5od\nS1f6XfCAd0WpikWZdbs2q20k5KQDQIH3KtujmPTGx3rZxq0Y9xCa30znPZk/EA2c\nVUhz7cYgyH+dN86Iu8EIzqLsjMOMJz/rRDOr1FL2fM+KyhYfxY8ps6AVHiSgqYV+\n0wXOioWyrwKBgQC+XQ5HwUf1pGou6RAzjADAd1ZStSt+X75NSV4eaUIhfmlsguPc\nM8tgOzFFKRV/4A801WbY8dUUud9Sb7b7sh10VJ3hw//z851Gv48KkpmzXkLdIXtX\n+rL3ugGBt0AT7TY/APOW5h983iNSeUwpk2e1PrOtqg8xZ8Zv2f6MUv/KOQKBgCti\nZTietaOneEHopG0VkTYzLnY1Bth+3hGSaWiemUMRvADrRI9aRlJcaPyhZ1H5s8mz\nurOToO9xJnm24GUqDGa5oV6oHqRnWbnX2A+NlGYQiwlTQlBF4XXQy6gEMMnZiVPN\njFVnebJl/hs2XI/nikYaTCJPiicHcldmY9z+AZOfAoGBANSgx/zKIs0CietLSTD2\n66ncJukfA5U1ZKz3Kvnacjp13oBea85V6xooFHWQASoiV4sEybcviDQdPehISrgY\no7XiHjOWsVVSx2PDRJT38qw26vlFs2LtYpf1x8by8YTsrkKvfU1qKsZqyqtYbS/X\n5qCQ4iJ6+wkHTzpqSqLZ2kHs\n-----END PRIVATE KEY-----\n",
+  "client_email": "firebase-adminsdk-tgl92@montaillou-bdd.iam.gserviceaccount.com",
+  "client_id": "112544598871622022690",
+  "auth_uri": "https://accounts.google.com/o/oauth2/auth",
+  "token_uri": "https://oauth2.googleapis.com/token",
+  "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
+  "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/firebase-adminsdk-tgl92%40montaillou-bdd.iam.gserviceaccount.com",
+  "universe_domain": "googleapis.com"
+}