version7

app.py

@@ -16,21 +16,13 @@ def extract_text_from_pdf(pdf_file):
     
     pdf_text = ""
     try:
-        # Save the uploaded file to a temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_pdf:
-            temp_pdf.write(pdf_file)
-            temp_path = temp_pdf.name
-        
-        # Open the temporary file with PyPDF2
-        with open(temp_path, 'rb') as f:
+        # In Hugging Face Spaces, pdf_file is already a file path
+        with open(pdf_file.name, 'rb') as f:
             pdf_reader = PyPDF2.PdfReader(f)
             for page_num in range(len(pdf_reader.pages)):
                 page = pdf_reader.pages[page_num]
                 pdf_text += page.extract_text() + "\n"
         
-        # Clean up the temporary file
-        os.unlink(temp_path)
-        
     except Exception as e:
         return f"Error processing PDF: {str(e)}"
     
@@ -122,22 +114,6 @@ def query_qa_system(question, model, index, text_chunks, similarity_threshold=0.
             'found_answer': False
         }
 
-def ask_question(question, model, index, text_chunks):
-    """
-    User-friendly interface for asking questions
-    """
-    result = query_qa_system(question, model, index, text_chunks)
-    print("\nQ:", question)
-    print("-" * 50)
-    if result['found_answer']:
-        print(f"Found matching section (confidence: {result['confidence']:.2f}):")
-        print(f"\n{result['full_text']}\n")
-        return result
-    else:
-        print(result['full_text'])
-        print(f"Best match confidence: {result['confidence']:.2f}")
-        return result
-
 # Global variables to store model, index, and text chunks
 global_model = None
 global_index = None