Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import os
-from pathlib import Path 
-
-api_token = os.getenv("HF_TOKEN")
-
+from pathlib import Path  # Adicionando a importação necessária
+import re
+from unidecode import unidecode
+import chromadb
 from langchain_community.vectorstores import FAISS
 from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -16,6 +16,8 @@ from langchain.memory import ConversationBufferMemory
 from langchain_community.llms import HuggingFaceEndpoint
 import torch
 
+api_token = os.getenv("HF_TOKEN")
+
 list_llm = ["meta-llama/Meta-Llama-3-8B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"]  
 list_llm_simple = [os.path.basename(llm) for llm in list_llm]
 
@@ -58,18 +60,18 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
     if llm_model == "meta-llama/Meta-Llama-3-8B-Instruct":
         llm = HuggingFaceEndpoint(
             repo_id=llm_model,
-            huggingfacehub_api_token = api_token,
-            temperature = temperature,
-            max_new_tokens = max_tokens,
-            top_k = top_k,
+            huggingfacehub_api_token=api_token,
+            temperature=temperature,
+            max_new_tokens=max_tokens,
+            top_k=top_k,
         )
     else:
         llm = HuggingFaceEndpoint(
-            huggingfacehub_api_token = api_token,
+            huggingfacehub_api_token=api_token,
             repo_id=llm_model, 
-            temperature = temperature,
-            max_new_tokens = max_tokens,
-            top_k = top_k,
+            temperature=temperature,
+            max_new_tokens=max_tokens,
+            top_k=top_k,
         )
     
     progress(0.75, desc="Defining buffer memory...")
@@ -94,7 +96,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
 # Generate collection name for vector database
 def create_collection_name(filepath):
     collection_name = Path(filepath).stem
-    collection_name = collection_name.replace(" ","-") 
+    collection_name = collection_name.replace(" ", "-") 
     collection_name = unidecode(collection_name)
     collection_name = re.sub('[^A-Za-z0-9]+', '-', collection_name)
     collection_name = collection_name[:50]
@@ -250,4 +252,4 @@ def demo():
 
 
 if __name__ == "__main__":
-    demo()
+    demo()