Update app.py

app.py

@@ -26,6 +26,15 @@ client = InferenceClient(
 )
 
 
+
+def read_txt(txt_path):
+    text=""
+    with open(txt_path,"r") as f:
+        text = f.read()
+    f.close()
+    print (text)
+    return text
+
 def read_pdf(pdf_path):
     from pypdf import PdfReader
     text=""
@@ -179,21 +188,27 @@ def summarize(inp,history,data=None,file=None):
                 zz=read_pdf(file)
                 print (zz)
                 data=f'{data}\nFile:\n{zz}'
+            elif file.endswith(".txt"):
+                zz=read_txt(file)
+                print (zz)
+                data=f'{data}\nFile:\n{zz}'                
         except Exception as e:
+            data = "Error"
             print (e) 
-    
-    print(inp)
-    out = str(data)
-    rl = len(out)
-    print(f'rl:: {rl}')
-    c=0
-    for i in str(out):
-        if i == " " or i=="," or i=="\n":
-            c +=1
-    print (f'c:: {c}')
-    
-    rawp = compress_data(c,inp,out)    
-    
+    if not data == "Error":
+        print(inp)
+        out = str(data)
+        rl = len(out)
+        print(f'rl:: {rl}')
+        c=0
+        for i in str(out):
+            if i == " " or i=="," or i=="\n":
+                c +=1
+        print (f'c:: {c}')
+        
+        rawp = compress_data(c,inp,out)    
+    else:
+        rawp = "Error"
     #print (rawp)
     #print (f'out:: {out}')
     #history += "observation: the search results are:\n {}\n".format(out)