app.py

import gradio as gr
from gradio_client import Client
import PyPDF2

# Função para extrair texto de um PDF
def extract_text_from_pdf(pdf_path):
    with open(pdf_path, 'rb') as file:
        reader = PyPDF2.PdfFileReader(file)
        text = ''
        for page_num in range(reader.numPages):
            page = reader.getPage(page_num)
            text += page.extract_text()
    return text

# Função para gerar parecer usando o modelo de linguagem
def generate_analysis(text):
    client = Client("yuntian-deng/ChatGPT")
    result = client.predict(
        inputs=text,
        top_p=1,
        temperature=1,
        chat_counter=0,
        chatbot=[],
        api_name="/predict"
    )
    return result

# Função principal para a interface
def analyze_pdf(pdf_file):
    text = extract_text_from_pdf(pdf_file.name)
    analysis = generate_analysis(text)
    return analysis

# Interface Gradio
iface = gr.Interface(
    fn=analyze_pdf,
    inputs=gr.File(label="Upload PDF"),
    outputs=gr.Textbox(label="Parecer Gerado"),
    title="Sistema de Análise de PDF com RAG",
    description="Faça upload de um PDF para gerar um parecer."
)

iface.launch()