Update app.py

app.py

@@ -10,6 +10,7 @@ from langchain.chains import RetrievalQA
 import os
 import urllib.request
 import ssl
+import bitsandbytes
 import tempfile
 import mimetypes
 from dotenv import load_dotenv
@@ -127,7 +128,7 @@ def embed_pdf(file, filename, collection_name, file_type):
             context = ssl._create_unverified_context()
             with urllib.request.urlopen(file, context=context) as response, open(filename, 'wb') as out_file:
                 data = response.read()
-                out_file.write(data_bytes)
+                out_file.write(data)
             file_path = filename
         except Exception as e:
             return {"error": f"Error downloading file from URL: {e}"}
@@ -139,7 +140,7 @@ def embed_pdf(file, filename, collection_name, file_type):
         file_content = file
         file_path = os.path.join('./', filename)
         with open(file_path, 'wb') as f:
-            f.write(file_bytes)
+            f.write(file)
     else:
         return {"error": "Invalid file type"}
 
@@ -150,18 +151,14 @@ def embed_pdf(file, filename, collection_name, file_type):
     docs = loader.load()
 
     # Generate embeddings and store documents in Weaviate
-    embeddings = CohereEmbeddings(model="embed-multilingual-v2.0", cohere_api_key=cohere_api_key)
     for doc in docs:
-        embedding = embeddings.embed([doc['text']])
+        embedding = vectorstore.embedding.embed([doc['text']])
         weaviate_document = {
             "text": doc['text'],
             "embedding": embedding
         }
         client.data_object.create(data_object=weaviate_document, class_name=collection_name)
 
-    # Clean up if a temporary file was created
-    if isinstance(file, bytes):
-        os.remove(file_path)
     return {"message": f"Documents embedded in Weaviate collection '{collection_name}'"}
 
 def retrieve_info(query):