Update app.py

app.py

@@ -222,9 +222,10 @@ def create_audio_recorder():
 class AdvancedRAG:
     def __init__(self):
         self.thread_id: Optional[str] = None
+        self.file_ids: List[str] = []
         self.assistant_id: Optional[str] = os.getenv("ASSISTANT_ID")
-        if not self.assistant_id:
-            raise ValueError("ASSISTANT_ID environment variable is not set")
+        if hasattr(self, 'vector_store_id'):
+            self.vector_store_id = None
 
     def create_thread(self) -> str:
         thread = openai.beta.threads.create()
@@ -232,49 +233,35 @@ class AdvancedRAG:
         return self.thread_id
 
     def upload_document(self, file) -> str:
+        # Always reset state for a new document
+        self.thread_id = None
+        self.file_ids = []
+        if hasattr(self, 'vector_store_id'):
+            self.vector_store_id = None
         try:
-            if file is None:
-                raise ValueError("No file provided")
-
-            # The file is now a bytes object directly
-            file_bytes = file
-            if not file_bytes:
-                raise ValueError("Empty file")
-
-            # Create a temporary file with a generic extension
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".txt") as tmp:
-                tmp.write(file_bytes)
+            # Create a new file
+            with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.name)[1]) as tmp:
+                tmp.write(file.read())
                 tmp.flush()
-                
-                # Upload the file to OpenAI
                 with open(tmp.name, "rb") as file_obj:
                     file_obj = openai.files.create(
                         file=file_obj,
                         purpose="assistants"
                     )
-
-            # Create a new thread if one doesn't exist
-            if not self.thread_id:
-                self.create_thread()
-
-            # Add the file to the thread with the required tools object
+                    self.file_ids.append(file_obj.id)
+            # Create a new thread for the new document
+            thread = openai.beta.threads.create()
+            self.thread_id = thread.id
+            # Add the file to the thread
             openai.beta.threads.messages.create(
                 thread_id=self.thread_id,
                 role="user",
                 content="I have uploaded a document. Please analyze it.",
                 attachments=[{"file_id": file_obj.id, "tools": [{"type": "file_search"}]}]
             )
-
             return file_obj.id
         except Exception as e:
             raise Exception(f"Error uploading document: {str(e)}")
-        finally:
-            # Clean up temporary file
-            if 'tmp' in locals():
-                try:
-                    os.unlink(tmp.name)
-                except:
-                    pass
 
     def ask_question(self, question: str) -> str:
         try:
@@ -428,8 +415,6 @@ with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
     """)
 
     chatbot = gr.Chatbot(height=400, elem_classes="gradio-chatbot", label=None, type="messages")
-    file_output = gr.Textbox(label="Upload Status", interactive=False, elem_classes="textbox")
-    question = gr.Textbox(label="Type your question and press Enter", placeholder="Ask a question about your document...", elem_classes="textbox")
     audio_input = gr.Audio(type="filepath", label="Record or Upload Audio", elem_classes="gradio-audio", visible=False)
     tts_output = gr.Audio(label="Assistant Voice Reply", interactive=False, visible=False)
 
@@ -443,9 +428,16 @@ with gr.Blocks(css=custom_css, title="Document Q&A System") as demo:
                 audio_input
                 send_voice_btn = gr.Button("Send Voice Note", elem_classes="send-btn", visible=False)
                 reset_btn = gr.Button("Reset Chat & Upload New Document", elem_classes="reset-btn")
-                file_output
-                question
+                file_output = gr.Textbox(label="Upload Status", interactive=False, elem_classes="textbox")
+                question = gr.Textbox(label="Type your question and press Enter", placeholder="Ask a question about your document...", elem_classes="textbox")
                 file_input.change(process_file, file_input, file_output)
+                def reset_all():
+                    rag.thread_id = None
+                    if hasattr(rag, 'file_ids'):
+                        rag.file_ids = []
+                    if hasattr(rag, 'vector_store_id'):
+                        rag.vector_store_id = None
+                    return "", [], "", None, None
                 reset_btn.click(reset_all, None, [file_output, chatbot, question, audio_input, tts_output])
                 def show_audio():
                     return {audio_input: gr.update(visible=True), send_voice_btn: gr.update(visible=True)}