File size: 1,039 Bytes
56d6b19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from openai import OpenAI
import gradio
import shutil
import pdfplumber
import os

def read_PDF_2(fileobj):
    print("readPDF called")
    pdf_text = ""

    with pdfplumber.open(fileobj) as pdf:
        for page in pdf.pages:
            pdf_text += page.extract_text()

    return pdf_text


def askPDF(upload_PDF, prompt):
    print("askPDF called")
    client = OpenAI()

    pdf_text = read_PDF_2(upload_PDF)
    systemMessage = '"""' + pdf_text + '"""' + \
        '\nAnswer all questions according to the text delimited in tripple quotes above'
    completion = client.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": systemMessage},
            {"role": "user", "content": prompt}
        ]
    )

    return (completion.choices[0].message).content


# gradio deployment
demo = gradio.Interface(fn=askPDF,
                        inputs=[gradio.File(file_types=['.pdf']), 'text'],
                        outputs="text",
                        )

demo.launch()