Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import faiss
 import numpy as np
+import fitz
+import docx
+from pptx import Presentation
 
 from sentence_transformers import SentenceTransformer
 from transformers import pipeline
@@ -21,14 +24,44 @@ gen_model=AutoModelForSeq2SeqLM.from_pretrained(gen_model_name)
 chunks_store=[]
 index=None
 
+def extract_text(file):
 
-def build_kb(text):
+    name=file.name.lower()
+
+    text=""
+
+    if name.endswith(".pdf"):
+        pdf=fitz.open(file.name)
+        for page in pdf:
+            text += page.get_text()
+
+    elif name.endswith(".docx"):
+        doc=docx.Document(file.name)
+        for p in doc.paragraphs:
+            text += p.text + "\n"
+
+    elif name.endswith(".pptx"):
+        prs=Presentation(file.name)
+
+        for slide in prs.slides:
+            for shape in slide.shapes:
+                if hasattr(shape,"text"):
+                    text += shape.text + "\n"
+
+    else:
+        text="Unsupported file format"
+
+    return text
+    
+def build_kb(file):
 
     global chunks_store,index
 
+    text=extract_text(file)
+
     chunks=[
-        text[i:i+300]
-        for i in range(0,len(text),300)
+        text[i:i+500]
+        for i in range(0,len(text),500)
     ]
 
     chunks_store=chunks
@@ -38,12 +71,12 @@ def build_kb(text):
     dim=embeddings.shape[1]
 
     index=faiss.IndexFlatL2(dim)
+
     index.add(
-        np.array(embeddings).astype("float32")
+      np.array(embeddings).astype("float32")
     )
 
-    return "Knowledge Base Created"
-
+    return f"Knowledge Base Created with {len(chunks)} chunks"
 
 def ask_question(question):
 
@@ -107,9 +140,8 @@ Ask questions over your own documents
 
     with gr.Tab("Build Knowledge Base"):
 
-        doc=gr.Textbox(
-            lines=12,
-            label="Paste Document Text"
+        doc=gr.File(
+            label="Upload PDF / DOCX / PPTX"
         )
 
         status=gr.Textbox(label="Status")