app.py

import google.generativeai as genai
import gradio as gr
from pypdf import PdfReader

def pdfSummarizer(gemini_api_key, pdf_file, kind):

    gemini_api_key = gemini_api_key

    genai.configure(api_key = gemini_api_key)

    model = genai.GenerativeModel("gemini-1.5-pro")

    pdf_file = PdfReader(pdf_file)

    pdf_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"])

    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?"])

    elif kind == "Sentence":
      response = model.generate_content([pdf_text, "can you summarize this document in one sentence?"])

    return response.text

with gr.Blocks(
    theme="gstaff/whiteboard",
    analytics_enabled=True,
    fill_height=True,
    css="#sidebar {background-color:black;color: white;}"
) as app:
    gr.HTML("<div style='text-align:center;overflow:hidden;'><h2>&#128211; PDF Summarizer &#128211;</h2></div>")
    with gr.Sidebar(elem_id = "sidebar"):
        gr.HTML("<div style='text-align:center;overflow:hidden;'><h3>&#128211; PDF Summarizer &#128211;</h3></div>")
        gr.HTML("<br>")
        # load sweet alert module from cdn
        gr.HTML("""
        <script src='https://cdn.jsdelivr.net/npm/[email protected]/dist/sweetalert2.all.min.js'></script>
        <link href='https://cdn.jsdelivr.net/npm/[email protected]/dist/sweetalert2.min.css' rel='stylesheet'>
        """)
        
        # Button with Javascript Event
        gr.HTML("""
        <div style="text-align:center;">
            <button onclick=infoAlert()
                style="padding: 10px; background-color: black; color: white; border: none; border-radius: 5px; cursor: pointer;">
                Info
            </button>
            <script>
                function infoAlert(){
                    Swal.fire({
                    title: 'How To Use',
                    text: 'Enter your Gemini API key, upload PDF document, select summary kind, then hit the summarize button',
                    confirmButtonColor: 'black',
                    timer: 7000
                    });
                }
            </script>
        </div>
        """) 
        
        gr.HTML("<div style='text-align:center;overflow:hidden;'><h4>Gemini API Key</h4></div>")
        api_key = gr.Text(label = "",placeholder="Enter your Google Gemini API key here")
        
    with gr.Row():
        pdf_input = gr.File(file_types=[".pdf"])
        summary_type = gr.Radio(["5 Bullet Points","10 Bullet Points", "Paragraph", "Sentence"],value = "5 Bullet Points", label="Select Summary Kind")
        
    with gr.Row():
        btn = gr.Button("Summarize")
        clear_btn = gr.ClearButton(value="Clear")

    with gr.Column():
        gr.HTML("<div style='text-align:center;overflow:hidden;'><h2>Summary Output</h2></div>")
        out = gr.Markdown()

    btn.click(fn=pdfSummarizer, inputs=[api_key, pdf_input, summary_type], outputs=out)
    clear_btn.click(lambda: [None,None,"5 Bullet Points",""],inputs=None,outputs=[api_key, pdf_input, summary_type, out],queue=False)

app.launch(pwa=True)