Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import streamlit as st
 import PyPDF2
 import requests
-import numpy as np
 import faiss
 from groq import Groq
 
@@ -40,16 +39,16 @@ def chunk_text(text, max_length=500):
 def compute_embeddings(chunks):
     embeddings = []
     for chunk in chunks:
-        vector = np.array([ord(char) for char in chunk[:300]])  # Truncate to 300 characters
-        padded_vector = np.pad(vector, (0, 300 - len(vector)), mode="constant")
-        embeddings.append(padded_vector.astype(np.float32))
-    return np.array(embeddings)
+        vector = [ord(char) for char in chunk[:300]]  # Truncate to 300 characters
+        padded_vector = vector + [0] * (300 - len(vector))  # Zero-pad to fixed size
+        embeddings.append(padded_vector)
+    return embeddings
 
 # Function to create FAISS index
 def create_faiss_index(embeddings):
-    dimension = embeddings.shape[1]
+    dimension = len(embeddings[0])
     index = faiss.IndexFlatL2(dimension)
-    index.add(embeddings)
+    index.add(faiss.FloatVectorArray(embeddings))
     return index
 
 # Function to query Groq API
@@ -80,7 +79,7 @@ def main():
     if question:
         st.write("Searching for relevant chunks...")
         question_embedding = compute_embeddings([question])[0]
-        _, indices = index.search(np.array([question_embedding]), k=1)
+        _, indices = index.search(faiss.FloatVectorArray([question_embedding]), k=1)
         relevant_chunk = chunks[indices[0][0]]
 
         st.write("Generating answer using Groq API...")
@@ -91,4 +90,3 @@ def main():
 if __name__ == "__main__":
     main()
 
-