Update app.py

app.py

@@ -63,43 +63,6 @@ def get_text_embedding(text):
 doc_path = hf_hub_download(repo_id="xavierbarbier/rag_ngap", filename="resource/embeddings_ngap.faiss", repo_type="space")
 
 index = faiss.read_index(doc_path)
-"""
-reader = PdfReader( doc_path )
-
-
-text = []
-for p in np.arange(0, len(reader.pages), 1):
-    page = reader.pages[int(p)]
-
-    # extracting text from page
-    text.append(page.extract_text())
-
-text = ' '.join(text)
-
-chunks = [text[i:i + chunk_size] for i in range(0, len(text), chunk_size)]
-
-text_embeddings = np.array([get_text_embedding(chunk) for chunk in chunks])
-
-d = text_embeddings.shape[1]
-index = faiss.IndexFlatL2(d)
-index.add(text_embeddings)"""
-
-
-def extract_text(file):
-
-    
-    reader = PdfReader(file)
-
-    text = []
-    for p in np.arange(0, len(reader.pages), 1):
-        page = reader.pages[int(p)]
-
-        # extracting text from page
-        text.append(page.extract_text())
-
-    text = ' '.join(text)
-
-    return text
 
 def qa(question):
 
@@ -122,8 +85,6 @@ def qa(question):
     
     return prompt
 
-def test_func(text):
-    return len(index)
 
 with gr.Blocks() as demo:
     
@@ -133,7 +94,7 @@ with gr.Blocks() as demo:
     promp_output = gr.Textbox(label="prompt")
 
     
-    qa_button.click(test_func, question_input, promp_output)
+    qa_button.click(qa, question_input, promp_output)