Update app.py

app.py

@@ -2,10 +2,13 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import os
-import spaces
+import csv
+from datetime import datetime
+
 # === CONFIGURATION ===
-MODEL_NAME = "facebook/MobileLLM-Pro"   # ✅ Plus léger et rapide sur CPU
+MODEL_NAME = "facebook/MobileLLM-Pro"  # ✅ Léger et rapide sur CPU
 INFO_FILE = "infos_medicaux.txt"
+LOG_FILE = "historique_questions.csv"
 MAX_TOKENS = 800
 TEMPERATURE = 0.6
 
@@ -16,6 +19,7 @@ model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.float32
 )
+model.eval()
 print("✅ Modèle chargé avec succès")
 
 # === CHARGEMENT DU CONTEXTE MÉDICAL ===
@@ -27,7 +31,16 @@ else:
     medical_context = ""
     print("⚠️ Aucun fichier infos_medicaux.txt trouvé.")
 
-@spaces.GPU()
+# === FONCTION DE LOG DES QUESTIONS ===
+def log_question(user_message, response):
+    file_exists = os.path.exists(LOG_FILE)
+    with open(LOG_FILE, "a", encoding="utf-8", newline="") as csvfile:
+        writer = csv.writer(csvfile)
+        if not file_exists:
+            writer.writerow(["Timestamp", "Question", "Réponse"])
+        writer.writerow([datetime.now().isoformat(), user_message, response])
+
+# === FONCTION DE CHAT ===
 def chat_with_finanfa(message, history=None):
     if history is None:
         history = []
@@ -48,7 +61,7 @@ def chat_with_finanfa(message, history=None):
         conversation += f"Utilisateur : {user_msg}\nAssistant : {bot_msg}\n"
     conversation += f"Utilisateur : {message}\nAssistant :"
 
-    # Génération
+    # Génération de la réponse
     inputs = tokenizer(conversation, return_tensors="pt", truncation=True)
     with torch.no_grad():
         outputs = model.generate(
@@ -61,6 +74,10 @@ def chat_with_finanfa(message, history=None):
 
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     response = response.split("Assistant :")[-1].strip()
+
+    # Enregistrement automatique dans le CSV
+    log_question(message, response)
+
     return response
 
 # === INTERFACE GRADIO ===