Upload app.py

app.py

@@ -0,0 +1,65 @@
+# !pip install PyMuPDF gradio openai
+
+import fitz, openai
+import gradio as gr
+
+def formatar(texto):
+    texto = texto.replace('\n', ' ')
+    return texto
+
+def pdf_para_lista(documento):
+    conteudo = fitz.open(documento)
+    total_paginas = conteudo.page_count
+    lista_conteudo = []
+    # lê da primeira página à última
+    for i in range(0, total_paginas):
+        # pega todo o texto da página
+        texto = conteudo.load_page(i).get_text("text")
+        texto = formatar(texto)
+        # adiciona o indicador de numeração da páginas
+        pagina = 'Página '+str(i+1)+' - '
+        lista_conteudo.append(pagina+texto)
+
+    conteudo.close()
+    return lista_conteudo
+
+def getResposta(arquivo, pergunta, openAI_key):
+  openai.api_key = openAI_key
+  base_texto = pdf_para_lista(arquivo)
+  prompt = ""
+  # adiciona uma quebra de linha para cada página
+  for c in base_texto:
+      prompt += c + '\n\n'
+  # instancia o modelo GPT
+  completion = openai.ChatCompletion.create(
+  model = 'gpt-3.5-turbo',
+  messages = [
+    {'role': 'assistant', 'content': prompt},
+    {'role': 'user', 'content': pergunta},
+    {'role': 'system', 'content':'Retorne apenas texto contido no resultado. Retorne o número da página de cada resultado no formato [Página].'}
+  ],
+  temperature = 0
+  )
+  resposta = completion['choices'][0]['message']['content']
+  return resposta
+
+with gr.Blocks() as demo:
+
+    gr.Markdown(f'<center><h1>Aplicação IA Generativa em Documentos PDF</h1></center>')
+
+    with gr.Row():
+
+        with gr.Group():
+            gr.Markdown(f'<p style="text-align:center">Open AI API key <a href="https://platform.openai.com/account/api-keys">aqui</a></p>')
+            openAI_key=gr.Textbox(label='Sua chave OpenAI:')
+            arquivo = gr.File(label='Insira seu PDF:', file_types=['.pdf'])
+            pergunta = gr.Textbox(label='Faça um pergunta em relação ao PDF:')
+            btn = gr.Button(value='Enviar')
+            btn.style(full_width=True)
+        with gr.Group():
+            resposta = gr.Textbox(label='Resposta:')
+
+        btn.click(getResposta, inputs=[arquivo, pergunta, openAI_key], outputs=[resposta])
+
+demo.launch()
+