Upload chat_log.py with huggingface_hub

chat_log.py

@@ -0,0 +1,44 @@
+import csv
+from datetime import datetime
+import os
+
+class ChatLogger:
+    def __init__(self, csv_file='chats.csv'):
+        self.csv_file = csv_file
+        self.current_chat_id = None
+        self.check_csv_file()
+    
+    def check_csv_file(self):
+        """Crear archivo CSV si no existe"""
+        if not os.path.exists(self.csv_file):
+            with open(self.csv_file, 'w', newline='', encoding='utf-8') as f:
+                writer = csv.writer(f)
+                writer.writerow(['ChatID', 'Timestamp', 'Modo', 'Modelo_IA', 'Rol', 'Mensaje'])
+    
+    def start_new_chat(self, mode, ai_model):
+        """Iniciar nueva conversación"""
+        self.current_chat_id = datetime.now().strftime("%Y%m%d_%H%M%S")
+        return self.current_chat_id
+    
+    def log_message(self, message, is_user=False, mode=None, ai_model=None):
+        """Registrar mensaje en CSV"""
+        if not self.current_chat_id:
+            self.current_chat_id = self.start_new_chat(mode, ai_model)
+            
+        try:
+            with open(self.csv_file, 'a', newline='', encoding='utf-8') as f:
+                writer = csv.writer(f)
+                writer.writerow([
+                    self.current_chat_id,
+                    datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                    mode,
+                    ai_model,
+                    'Usuario' if is_user else 'Asistente',
+                    message.replace('\n', ' ')
+                ])
+        except Exception as e:
+            print(f"Error guardando mensaje en log: {e}")
+    
+    def end_chat(self):
+        """Finalizar chat actual"""
+        self.current_chat_id = None