Update app.py

app.py

@@ -243,34 +243,41 @@ class AdvancedRAG:
         self.thread_id = None
         self.file_ids = []
         if hasattr(self, 'vector_store_id'):
+            try:
+                openai.beta.vector_stores.delete(self.vector_store_id)
+            except Exception as e:
+                print(f"Warning: Could not delete vector store: {e}")
             self.vector_store_id = None
-        try:
-            # file is a bytes object, so assign a default name and extension
-            if not file:
-                raise Exception("No file uploaded.")
-            filename = 'uploaded_file.pdf'  # Default extension, or infer from UI if needed
-            with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(filename)[1]) as tmp:
-                tmp.write(file)
-                tmp.flush()
-                with open(tmp.name, "rb") as file_obj:
-                    file_obj = openai.files.create(
-                        file=file_obj,
-                        purpose="assistants"
-                    )
-                    self.file_ids = [file_obj.id]  # Only the new file
-            # Create a new thread for the new document
-            thread = openai.beta.threads.create()
-            self.thread_id = thread.id
-            # Send a message in the new thread with only the new file as an attachment
-            openai.beta.threads.messages.create(
-                thread_id=self.thread_id,
-                role="user",
-                content="I have uploaded a document. Please analyze it.",
-                attachments=[{"file_id": self.file_ids[0], "tools": [{"type": "file_search"}]}]
-            )
-            return self.file_ids[0]
-        except Exception as e:
-            raise Exception(f"Error uploading document: {str(e)}")
+
+        # Wait a moment to ensure deletion is processed
+        time.sleep(2)
+
+        # Upload new file
+        if not file:
+            raise Exception("No file uploaded.")
+        filename = 'uploaded_file.pdf'
+        with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(filename)[1]) as tmp:
+            tmp.write(file)
+            tmp.flush()
+            with open(tmp.name, "rb") as file_obj:
+                file_obj = openai.files.create(
+                    file=file_obj,
+                    purpose="assistants"
+                )
+                self.file_ids = [file_obj.id]
+
+        # Create a new thread for the new document
+        thread = openai.beta.threads.create()
+        self.thread_id = thread.id
+
+        # Send a message in the new thread with only the new file as an attachment
+        openai.beta.threads.messages.create(
+            thread_id=self.thread_id,
+            role="user",
+            content="I have uploaded a document. Please analyze it.",
+            attachments=[{"file_id": self.file_ids[0], "tools": [{"type": "file_search"}]}]
+        )
+        return self.file_ids[0]
 
     def ask_question(self, question: str) -> str:
         try: