Update views/rag_med.py

views/rag_med.py

@@ -9,8 +9,23 @@ from sentence_transformers import SentenceTransformer
 import pickle
 from prompt_template import prompt
 
-# root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
-# sys.path.append(root_dir)
+# Função para encontrar o arquivo
+def find_file(filename):
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    root_dir = os.path.dirname(current_dir)
+    
+    possible_paths = [
+        os.path.join(current_dir, filename),
+        os.path.join(root_dir, filename),
+        os.path.join("/content", filename),  # Para ambientes como Colab
+        os.path.join("/home/user/app", filename)  # Caminho comum no Hugging Face
+    ]
+    
+    for path in possible_paths:
+        if os.path.exists(path):
+            return path
+    
+    raise FileNotFoundError(f"Não foi possível encontrar o arquivo: {filename}")
 
 # Initialize the messages list in the session state
 if "messages" not in st.session_state: