Update app.py

app.py

@@ -4,7 +4,8 @@ import torch
 import transformers
 from transformers import pipeline
 import re
-import html
+import HTML
+from pypdf import PdfReader
 
 def remove_references(text):
     text = re.sub(r'\[\d+\]', '', text) ##[ref]
@@ -15,6 +16,13 @@ def remove_references(text):
     return text
 
     
+def extract_text_from_pdf(file_path):
+  reader = PdfReader(file_path)
+  text = ""
+  for page in reader.pages:
+      text += page.extract_text() + "\n"
+
+      
 def model(model_name):
   tokenizer = AutoTokenizer.from_pretrained(model_name)
   model = AutoModelForQuestionAnswering.from_pretrained(model_name,return_dict = False)
@@ -46,7 +54,9 @@ theme = gr.themes.Soft().set(
     body_text_color_subdued='*body_text_color',
     body_text_color_subdued_dark='*chatbot_code_background_color'
 )
-
+with gr.Column(scale=0.15, min_width=0):
+            btn = gr.UploadButton("📁", file_types=[".pdf",".csv",".doc"])
+    
 app = gr.Interface(fn=qa_result,
                    inputs = ['textbox', 'text'],
                    outputs = 'textbox', title = 'Ողջու՛յն։ Ես քո արհեստական բանականությամբ օգնականն եմ ', theme = theme, description = 'Տու՛ր ինձ տեքստ, ու տեքստին վերաբերող հարցեր, ու ես կօգնեմ քեզ պատասխանել հարցերին')