Update app.py

app.py

@@ -18,26 +18,26 @@ def get_gemini_response(input,pdf_cotent,prompt):
 
 def input_pdf_setup(uploaded_file):
     if uploaded_file is not None:
-        ## Convert the PDF to image
-        images=fitz.open(stream=pdf_bytes, filetype="pdf")
+        # Read PDF as bytes
+        pdf_bytes = uploaded_file.read()
+        pdf_doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+        first_page = pdf_doc[0]  # Get first page of the PDF
+        pix = first_page.get_pixmap()
 
-        first_page=images[0]
-
-        # Convert to bytes
+        # Convert page to image bytes
         img_byte_arr = io.BytesIO()
-        first_page.save(img_byte_arr, format='JPEG')
+        img_byte_arr.write(pix.tobytes("jpeg"))
         img_byte_arr = img_byte_arr.getvalue()
 
         pdf_parts = [
             {
                 "mime_type": "image/jpeg",
-                "data": base64.b64encode(img_byte_arr).decode()  # encode to base64
+                "data": base64.b64encode(img_byte_arr).decode()
             }
         ]
         return pdf_parts
     else:
         raise FileNotFoundError("No file uploaded")
-
 ## Streamlit App
 
 st.set_page_config(page_title="ATS Resume EXpert")