Update app.py

app.py

@@ -25,6 +25,18 @@ client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
 
+
+def read_pdf(pdf_path):
+    from pypdf import PdfReader
+
+    reader = PdfReader(f'{pdf_path}')
+    number_of_pages = len(reader.pages)
+    
+    page = reader.pages[:-1]
+    text = page.extract_text()
+    print (text)
+    return text
+
 def parse_action(string: str):
     print("PARSING:")
     print(string)
@@ -152,6 +164,8 @@ def compress_data(c, instruct, history):
 
 def summarize(inp,history,data=None,file=None):
     #file = None
+
+        
     if inp == "":
         inp = "Process this data"
         
@@ -160,11 +174,11 @@ def summarize(inp,history,data=None,file=None):
     yield "",history
     if file:
         try: 
-            with open(file,"r") as f:
-                zz=f.readlines()
-                f.close
-            print (zz)
-            data=f'{data}\nFile:\n{zz}'
+            print (file)
+            if file.endswith(".pdf"):
+                zz=read_pdf(file)
+                print (zz)
+                data=f'{data}\nFile:\n{zz}'
         except Exception as e:
             print (e)