Alejo760 commited on
Commit
011e236
verified
1 Parent(s): bfaa6de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -26,10 +26,6 @@ vectordb_guia = Chroma(persist_directory="./vector_db_guia2", embedding_function
26
  # Prepare the list of vector databases
27
  vectordb_list = [vectordb_PEI, vectordb_guia]
28
 
29
- # Si el usuario ha subido documentos personalizados, incluir su base de datos vectorial
30
- if 'vectordb_custom' in st.session_state:
31
- vectordb_list.append(st.session_state['vectordb_custom'])
32
-
33
  client = Groq(api_key=api_key)
34
 
35
  # Inicializar el modelo de chat
@@ -69,10 +65,6 @@ if uploaded_files:
69
  if st.button("A帽adir"):
70
  with st.spinner("Procesando documentos..."):
71
  # Crear una base de datos vectorial en memoria
72
- client_settings = Settings(
73
- chroma_api_impl="chromadb.api.local.LocalAPI",
74
- chroma_db_impl="chromadb.db.impl.sqlite.SqliteDB",
75
- )
76
  vectordb_custom = Chroma(
77
  embedding_function=embeddings,
78
  client_settings=client_settings,
@@ -97,11 +89,16 @@ if uploaded_files:
97
  # A帽adir los textos a la base de datos vectorial
98
  vectordb_custom.add_texts(texts)
99
 
 
 
100
  st.success("Documentos a帽adidos y procesados correctamente.")
101
 
102
  # Guardar la base de datos vectorial en la sesi贸n
103
  st.session_state['vectordb_custom'] = vectordb_custom
104
 
 
 
 
105
 
106
 
107
  def mejorar_texto_con_IA(var_name, var_value, vectordb_list, nombre_curso):
 
26
  # Prepare the list of vector databases
27
  vectordb_list = [vectordb_PEI, vectordb_guia]
28
 
 
 
 
 
29
  client = Groq(api_key=api_key)
30
 
31
  # Inicializar el modelo de chat
 
65
  if st.button("A帽adir"):
66
  with st.spinner("Procesando documentos..."):
67
  # Crear una base de datos vectorial en memoria
 
 
 
 
68
  vectordb_custom = Chroma(
69
  embedding_function=embeddings,
70
  client_settings=client_settings,
 
89
  # A帽adir los textos a la base de datos vectorial
90
  vectordb_custom.add_texts(texts)
91
 
92
+
93
+
94
  st.success("Documentos a帽adidos y procesados correctamente.")
95
 
96
  # Guardar la base de datos vectorial en la sesi贸n
97
  st.session_state['vectordb_custom'] = vectordb_custom
98
 
99
+ # Si el usuario ha subido documentos personalizados, incluir su base de datos vectorial
100
+ if 'vectordb_custom' in st.session_state:
101
+ vectordb_list.append(st.session_state['vectordb_custom'])
102
 
103
 
104
  def mejorar_texto_con_IA(var_name, var_value, vectordb_list, nombre_curso):