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

api_key = os.getenv("api_key")



genai.configure(api_key= api_key)
model = genai.GenerativeModel('gemini-1.5-pro-latest')

def get_summary_from_text(text, prompt):
    response = model.generate_content(f"{prompt}: {text}")
    return response.text

def extract_text_from_pdf(pdf_file):
    # Open the PDF file
    with open(pdf_file.name, 'rb') as file:
        text = ""
        reader = PdfReader(pdf_file)
        num_pages = len(reader.pages)
        for i in range(num_pages):
          page = reader.pages[i]
          text += page.extract_text()
    return text

def get_summary_from_pdf(pdf_file, prompt):
    text = extract_text_from_pdf(pdf_file)
    summary = get_summary_from_text(text, prompt)
    return summary

iface = gr.Interface(fn=get_summary_from_pdf, inputs=["file", "text"], outputs="text", title="Quarterly Info Summarizer by Farhan")
iface.launch(debug = True)