Update app.py

app.py

@@ -82,17 +82,26 @@ def pdf_chat(pdf_file, text_query, temperature, top_p, max_output_tokens):
     if pdf_file is None or not text_query:
         return "Please upload a PDF and provide a query."
 
-    # Extract text from all pages
-    doc = fitz.open(pdf_file)
-    text = "\n".join([page.get_text("text") for page in doc])  # Extract text from all pages
-
-    messages = [
-        {"role": "user", "content": [
-            {"type": "text", "text": text},  # Fixed format
-            {"type": "text", "text": text_query}
-        ]},
-    ]
-    return query_openai(messages, temperature, top_p, max_output_tokens)
+    try:
+        # Extract text from all pages of the PDF
+        doc = fitz.open(pdf_file.name)
+        text = "\n".join([page.get_text("text") for page in doc])  # Extract text from all pages
+
+        # If no text found, return an error
+        if not text.strip():
+            return "No text found in the PDF."
+
+        # Create the query message with the extracted text and the user's query
+        messages = [
+            {"role": "user", "content": [
+                {"type": "text", "text": text},  # The extracted text from the PDF
+                {"type": "text", "text": text_query}
+            ]},
+        ]
+        return query_openai(messages, temperature, top_p, max_output_tokens)
+    
+    except Exception as e:
+        return f"Error processing the PDF: {str(e)}"
 
 # Function to transcribe audio to text using OpenAI Whisper API
 def transcribe_audio(audio_binary, openai_api_key):
@@ -201,4 +210,4 @@ with gr.Blocks() as demo:
 
 # Launch Gradio App
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()