Update app.py

app.py

@@ -76,6 +76,7 @@ class RAGSystem:
         self.tokenizer = None
         self.model = None
         self.is_initialized = False
+        self.processed_files = set()  # Mantener registro de archivos procesados
     
     def initialize_model(self):
         """Initialize the base model and tokenizer."""
@@ -134,10 +135,20 @@ class RAGSystem:
         """Process uploaded documents and update the vector store."""
         try:
             documents = []
+            new_files = []
+            
+            # Procesar solo archivos nuevos
             for file in files:
-                docs = DocumentLoader.load_file(file.name)
-                documents.extend(docs)
+                if file.name not in self.processed_files:
+                    docs = DocumentLoader.load_file(file.name)
+                    documents.extend(docs)
+                    new_files.append(file.name)
+                    self.processed_files.add(file.name)
             
+            if not new_files:
+                logger.info("No new documents to process")
+                return
+                
             if not documents:
                 raise ValueError("No documents were successfully loaded.")
             
@@ -225,9 +236,10 @@ def process_response(user_input: str, chat_history: List, files: List) -> tuple:
         if not rag_system.is_initialized:
             rag_system.initialize_model()
         
-        if files and (rag_system.vector_store is None):
+        # Siempre procesar documentos si hay archivos nuevos
+        if files:
             rag_system.process_documents(files)
-        
+            
         response = rag_system.generate_response(user_input)
         
         # Clean and format response
@@ -345,8 +357,14 @@ try:
         def submit(user_input, chat_history, files):
             return process_response(user_input, chat_history, files)
             
+        def clear_context():
+            # Limpiar el historial y reiniciar el sistema
+            rag_system.vector_store = None
+            rag_system.processed_files.clear()
+            return None
+            
         message.submit(submit, [message, chatbot, files], [chatbot])
-        clear.click(lambda: None, None, chatbot)
+        clear.click(clear_context, None, chatbot)
 
     logger.info("Gradio interface created successfully")
     demo.launch()