Update app.py

app.py

@@ -4,16 +4,12 @@ from io import BytesIO
 import pypdf
 import os
 
-# **IMPORTANT:** Replace this with your actual Hugging Face Space URL
-# Example: "https://ruslanmv-milvus-server.hf.space" 
-#           "https://huggingface.co/spaces/ruslanmv/Milvus-Server/rag"
-#           "https://ruslanmv-milvus-server.hf.space/rag"
-#space_url = os.environ.get("SPACE_URL", "https://ruslanmv-milvus-server.hf.space")  # or set it here
-#rag_url = os.environ.get("SPACE_URL", "https://ruslanmv-milvus-server.hf.space") + "/rag"
-#insert_url = os.environ.get("SPACE_URL", "https://ruslanmv-milvus-server.hf.space") + "/insert"
-space_url = os.environ.get("SPACE_URL", "https://ruslanmv-milvus-server.hf.space")
+# **IMPORTANT:** Set your Hugging Face Space URL here or as an environment variable
+space_url = os.environ.get("SPACE_URL", "https://ruslanmv-milvus-server.hf.space")  # Your Milvus Server Space URL
+rag_url = space_url + "/rag"
+insert_url = space_url + "/insert"
 
-# Function to extract text from a PDF file
+# Function to extract text from a PDF file (no changes needed here)
 def extract_text_from_pdf(pdf_file):
     pdf_stream = BytesIO(pdf_file)
     reader = pypdf.PdfReader(pdf_stream)
@@ -24,12 +20,10 @@ 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):
-    #insert_url = server_url.rstrip("/") + "/insert"
-    insert_url = server_url + "/insert"
     try:
         files = {'file': (pdf_file.name, open(pdf_file.name, 'rb'), 'application/pdf')}
-        response = requests.post(insert_url, files=files, timeout=600)  # Increased timeout
-        response.raise_for_status()  # Raise an exception for bad status codes
+        response = requests.post(insert_url, files=files, timeout=600)
+        response.raise_for_status()
         return "PDF uploaded and indexed successfully!"
     except requests.exceptions.RequestException as e:
         return f"Error during PDF upload: {e}"
@@ -38,8 +32,6 @@ def upload_and_index_pdf(pdf_file, server_url):
 
 # Function to perform RAG query
 def perform_rag_query(question, server_url):
-    #rag_url = server_url.rstrip("/") + "/rag"
-    rag_url = server_url + "/rag"
     try:
         response = requests.post(rag_url, json={"question": question}, timeout=300)
         response.raise_for_status()
@@ -50,6 +42,14 @@ def perform_rag_query(question, server_url):
     except Exception as e:
         return f"An unexpected error occurred: {e}"
 
+# Example questions
+example_questions = [
+    "What are the enabling technologies for GPT?",
+    "Explain the potential applications of GPT.",
+    "What are some emerging challenges with GPT technology?",
+    "Describe the future directions for GPT research."
+]
+
 # Gradio interface setup
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -60,20 +60,35 @@ with gr.Blocks() as demo:
     )
     with gr.Row():
         with gr.Column():
-            pdf_input = gr.File(label="Upload PDF", type="file")
+            pdf_input = gr.File(label="Upload PDF", type="filepath")  # Changed type to "filepath"
             server_url_input = gr.Textbox(
                 label="Milvus Server URL",
-                value=space_url,  # Use the environment variable or default
+                value=space_url,
                 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")
-    
+
     with gr.Row():
         with gr.Column():
             question_input = gr.Textbox(label="Ask a question about the PDF")
             query_button = gr.Button("Ask")
+            # Example questions
+            gr.Examples(
+                examples=example_questions,
+                inputs=question_input,
+                label="Example Questions",
+            )
         with gr.Column():
             answer_output = gr.Textbox(label="Answer")