shayanSolSameed commited on
Commit
56d6b19
·
1 Parent(s): 5da6762

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from openai import OpenAI
2
+ import gradio
3
+ import shutil
4
+ import pdfplumber
5
+ import os
6
+
7
+ def read_PDF_2(fileobj):
8
+ print("readPDF called")
9
+ pdf_text = ""
10
+
11
+ with pdfplumber.open(fileobj) as pdf:
12
+ for page in pdf.pages:
13
+ pdf_text += page.extract_text()
14
+
15
+ return pdf_text
16
+
17
+
18
+ def askPDF(upload_PDF, prompt):
19
+ print("askPDF called")
20
+ client = OpenAI()
21
+
22
+ pdf_text = read_PDF_2(upload_PDF)
23
+ systemMessage = '"""' + pdf_text + '"""' + \
24
+ '\nAnswer all questions according to the text delimited in tripple quotes above'
25
+ completion = client.chat.completions.create(
26
+ model="gpt-3.5-turbo",
27
+ messages=[
28
+ {"role": "system", "content": systemMessage},
29
+ {"role": "user", "content": prompt}
30
+ ]
31
+ )
32
+
33
+ return (completion.choices[0].message).content
34
+
35
+
36
+ # gradio deployment
37
+ demo = gradio.Interface(fn=askPDF,
38
+ inputs=[gradio.File(file_types=['.pdf']), 'text'],
39
+ outputs="text",
40
+ )
41
+
42
+ demo.launch()