Update app.py

app.py

@@ -1,35 +1,55 @@
 import google.generativeai as genai
 import gradio as gr
-import numpy as np
-import PIL.Image
+from pypdf import PdfReader
 
-genai.configure(api_key="AIzaSyA7tPavobVN5_3-BJ0qhFT5HVjO4V19QWk")
+def pdfSummarizer(gemini_api_key, pdf_file, kind):
 
-def ImageChat(image):
+    gemini_api_key = gemini_api_key
 
-  # load model
-  model = genai.GenerativeModel("gemini-1.5-flash")
+    genai.configure(api_key = gemini_api_key)
 
-  # check image file and convert to a Numpy array
-  if isinstance(image, np.ndarray):
+    model = genai.GenerativeModel("gemini-1.5-pro")
 
-        img = PIL.Image.fromarray(image)
-  else:
-        img = PIL.Image.open(image)
+    pdf_file = PdfReader(pdf_file)
 
-  response =  model.generate_content(["write a short story about the image", img])
+    pdf_text = ""
 
-  return response.text
+    for page in pdf_file.pages:
+        pdf_text += page.extract_text()
 
+    if kind == "5 Bullet Points":
+      response = model.generate_content([pdf_text, "can you summarize this document in 5 bullet points? Use bullet points and not asterisks"])
 
-app = gr.Interface(ImageChat,
-                   inputs = gr.Image(label = "Image"),
-                   outputs = gr.Text(label = "Story"),
-                   examples = ["rubiks cube.jpg","giraffe.jpg","street.jpg"],
-                   title = "Image-To-Story",
-                   theme = "patrickosornio/my_theme1")
+    elif kind == "10 Bullet Points":
+      response = model.generate_content([pdf_text, "can you summarize this document in 10 bullet points? Use bullet points and not asterisks"])
 
+    elif kind == "Paragraph":
+      response = model.generate_content([pdf_text, "can you summarize this document as a paragraph?"])
 
-app.launch()
+    elif kind == "Sentence":
+      response = model.generate_content([pdf_text, "can you summarize this document in one sentence?"])
+
+    return response.text
 
+import gradio as gr
 
+with gr.Blocks(
+    theme="gstaff/whiteboard",
+    title="PDF Summarizer",
+    analytics_enabled=True,
+    fill_height=True
+) as app:
+    gr.Markdown("<center><h2>PDF Summarizer</h2></center>")
+    with gr.Row():
+        inp = [
+            gr.Text(label="Gemini API Key", placeholder="Enter your Google Gemini API key here"),
+            gr.File(file_types=[".pdf"]),
+            gr.Radio(["5 Bullet Points","10 Bullet Points", "Paragraph", "Sentence"], label="Select Summary Kind")
+        ]
+        btn = gr.Button("Summarize")
+    with gr.Column():
+        out = gr.Text(label="Response")
+
+    btn.click(fn=pdfSummarizer, inputs=inp, outputs=out)
+
+app.launch()