Update app.py

app.py

@@ -55,6 +55,9 @@ uploads_dir = os.path.join(app.root_path,'static', 'uploads')
 
 os.makedirs(uploads_dir, exist_ok=True)
 
+defaultEmbeddingModelID = 0
+defaultLLMID=0
+
 
 def pretty_print_docs(docs):
     print(f"\n{'-' * 100}\n".join([f"Document {i + 1}:\n\n" + "Document Length>>>" + str(
@@ -241,13 +244,20 @@ def process_json():
 def file_Upload():
     fileprovided = not request.files.getlist('files[]')[0].filename == ''
     urlProvided = not request.form.getlist('weburl')[0] == ''
+    embeddingModelProvided = not request.form.getlist('embeddingModelID')[0] == ''
     print("*******")
     print("File Provided:" + str(fileprovided))
     print("URL Provided:" + str(urlProvided))
+    print("Embedding Model Provided:" + str(embeddingModelProvided))
     print("*******")
 
     print(uploads_dir)
     documents = loadKB(fileprovided, urlProvided, uploads_dir, request)
+    embeddingModelID = defaultEmbeddingModelID
+    if embeddingModelProvided:
+        embeddingModelID = int(request.form.getlist('embeddingModelID')[0])
+    global vectordb
+    vectordb = createVectorDB(documents, embeddingModelID)
     vectordb=createVectorDB(documents)
     return render_template("index.html")