Upload folder using huggingface_hub

README.md

@@ -1,12 +1,6 @@
 ---
-title: Ask PDF
-emoji: 👀
-colorFrom: pink
-colorTo: pink
+title: Ask_PDF
+app_file: gradio_chatwithpdf.py
 sdk: gradio
-sdk_version: 4.5.0
-app_file: app.py
-pinned: false
+sdk_version: 3.44.4
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

gradio_chatwithpdf.py

@@ -0,0 +1,53 @@
+
+from openai import OpenAI
+import gradio
+import shutil
+import pdfplumber
+import os
+
+
+def read_PDF(pdf_path):
+    pdf_text = ""
+    with pdfplumber.open(pdf_path) as pdf:
+        for page in pdf.pages:
+            pdf_text += page.extract_text()
+    return pdf_text
+
+
+def read_PDF_2(fileobj):
+    pdf_text = ""
+
+    path = "./" + os.path.basename(fileobj.name)
+    shutil.copyfile(fileobj.name, path)
+
+    with pdfplumber.open(path) as pdf:
+        for page in pdf.pages:
+            pdf_text += page.extract_text()
+
+    return pdf_text
+
+
+def askPDF(upload_PDF, prompt):
+    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()