Update app.py

app.py

@@ -1,45 +1,36 @@
-from PIL import Image
 import os
-import io
-import sys
 import google.generativeai as genai
-import fitz  # PyMuPDF
 import gradio as gr
 
 
 GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-1.5-flash-latest')
+model = genai.GenerativeModel('gemini-2.0-flash-exp')
 generation_config = genai.GenerationConfig(temperature=0)
 
 
 def upload(file):
     if not file:
         return "**ERROR!!! FILE NOT UPLOADED**"
-    
-    pics = []
-    doc = fitz.open(file)
-    for page_num in range(len(doc)):
-        page = doc.load_page(page_num)
-        pix = page.get_pixmap()
-        pics.append(Image.open(io.BytesIO(pix.tobytes())))
 
     prompt = "You are an expert in medical report analysis. Analyze the given medical report and generate a summarized report of the same."\
          "All the medical tests in the given medical report must be specified in the summarized report. The response generated should be well-structured with health tips based on the report."\
 
-    contents = [prompt]
-    contents.extend(pics)
+    file = genai.upload_file(file)
+    
+    contents = [prompt, file]
+    
     response = model.generate_content(contents)
 
     return response.text
 
 
 demo = gr.Interface(
-    fn=upload,
-    inputs=gr.File(file_types=['.pdf']),    # Input: File input
-    outputs=gr.Markdown(), # Output: Textbox to display the extracted text
-    title="Medical Report Summarizer",
-    description="Note: This summary is based on the provided medical report and does not constitute medical advice. Please consult with your healthcare provider for any questions or concerns regarding your health."
+    fn = upload,
+    inputs = gr.File(file_types=['.pdf']),    # Input: File input
+    outputs = gr.Markdown(), # Output: Textbox to display the extracted text
+    title = "Medical Report Summarizer",
+    description = "Note: This summary is based on the provided medical report and does not constitute medical advice. Please consult with your healthcare provider for any questions or concerns regarding your health."
 )