Update app.py

app.py

@@ -21,7 +21,12 @@ def extract_text_from_pdf(pdf_file):
 # Function to handle PDF upload and insertion into Milvus
 def upload_and_index_pdf(pdf_file, server_url):
     try:
-        files = {'file': (pdf_file.name, open(pdf_file.name, 'rb'), 'application/pdf')}
+        # Check if pdf_file is a file path (string) or a file object
+        if isinstance(pdf_file, str):
+            files = {'file': (pdf_file, open(pdf_file, 'rb'), 'application/pdf')}
+        else:
+            files = {'file': (pdf_file.name, pdf_file, 'application/pdf')}
+        
         response = requests.post(insert_url, files=files, timeout=600)
         response.raise_for_status()
         return "PDF uploaded and indexed successfully!"
@@ -67,15 +72,6 @@ with gr.Blocks() as demo:
                 placeholder="Enter your Milvus Server URL"
             )
             upload_button = gr.Button("Upload and Index PDF")
-
-            # Load and index the default PDF on startup (if it exists)
-            if os.path.exists("transformers.pdf"):  # Check if transformers.pdf exists
-                with open("transformers.pdf", "rb") as f:
-                    default_pdf_content = f.read()
-                # Using gr.State to store the default PDF content between sessions
-                default_pdf_file = gr.State(default_pdf_content)  
-                upload_and_index_pdf(default_pdf_file, server_url_input.value)
-
         with gr.Column():
             upload_output = gr.Textbox(label="Upload Status")
 
@@ -92,6 +88,15 @@ with gr.Blocks() as demo:
         with gr.Column():
             answer_output = gr.Textbox(label="Answer")
 
+    # Load and index the default PDF on startup (if it exists)
+    
+    if os.path.exists("transformers.pdf"):
+        print("transformers.pdf exists")
+        upload_and_index_pdf("transformers.pdf", space_url)
+        upload_output.value = "Default PDF (transformers.pdf) indexed on startup!"  # Update status
+    else:
+        print("transformers.pdf does not exist")
+
     upload_button.click(
         fn=upload_and_index_pdf,
         inputs=[pdf_input, server_url_input],