bgonpin commited on
Commit
f98e1a1
·
verified ·
1 Parent(s): 2fd40d5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. main gradio.py +12 -33
main gradio.py CHANGED
@@ -314,45 +314,24 @@ if __name__ == "__main__":
314
  print("🚀 Sistema RAG - Consulta de Documentos")
315
  print("=" * 50)
316
 
317
- # Preguntar al usuario qué acción realizar
318
- print("¿Qué deseas hacer?")
319
- print("1. Procesar documentos de las carpetas 'documents' y 'aportaciones' (recomendado si tienes documentos nuevos) 🚀")
320
- print("2. Pasar directamente al RAG (usar base de datos existente)")
321
-
322
- print("\n💡 ¡Novedad! Los usuarios ahora pueden subir documentos a la carpeta 'aportaciones' desde la interfaz web")
323
  print(" ¡Comparte tu conocimiento y enriquecer el sistema! 📚✨")
324
 
325
- while True:
326
- try:
327
- choice = input("\nElige una opción (1 o 2): ").strip()
328
- if choice == "1":
329
- print("\n📁 Procesando documentos...")
330
- process_documents = True
331
- break
332
- elif choice == "2":
333
- print("\n🚀 Pasando directamente al RAG...")
334
- process_documents = False
335
- break
336
- else:
337
- print("❌ Opción no válida. Por favor elige 1 o 2.")
338
- except KeyboardInterrupt:
339
- print("\n\n👋 ¡Hasta luego!")
340
- exit(0)
341
-
342
- # Inicializar sistema basado en la elección del usuario
343
  print("\nInicializando sistema...")
344
  db, embedding_model = initialize_system(process_documents)
345
 
346
- if process_documents:
347
- documents_processed = True
348
- print("✅ Sistema inicializado con documentos procesados")
 
349
  else:
350
- documents_processed = (db is not None)
351
- if documents_processed:
352
- print("✅ Sistema inicializado con documentos existentes")
353
- else:
354
- print("⚠️ No se pudo conectar a documentos existentes")
355
- print("💡 Sugerencia: Ejecuta el script con la opción 1 para procesar documentos")
356
 
357
  print("\n🚀 Iniciando interfaz web...")
358
  demo.launch(
 
314
  print("🚀 Sistema RAG - Consulta de Documentos")
315
  print("=" * 50)
316
 
317
+ # Usar siempre la opción de base de datos existente
318
+ print("💡 ¡Novedad! Los usuarios ahora pueden subir documentos a la carpeta 'aportaciones' desde la interfaz web")
 
 
 
 
319
  print(" ¡Comparte tu conocimiento y enriquecer el sistema! 📚✨")
320
 
321
+ print("\n🚀 Usando base de datos existente directamente...")
322
+ process_documents = False
323
+
324
+ # Inicializar sistema usando base de datos existente
 
 
 
 
 
 
 
 
 
 
 
 
 
 
325
  print("\nInicializando sistema...")
326
  db, embedding_model = initialize_system(process_documents)
327
 
328
+ # Verificar estado de inicialización
329
+ documents_processed = (db is not None)
330
+ if documents_processed:
331
+ print("✅ Sistema inicializado con documentos existentes")
332
  else:
333
+ print("⚠️ No se pudo conectar a documentos existentes")
334
+ print("💡 Asegúrate de que la base de datos ChromaDB esté disponible en 'chroma/'")
 
 
 
 
335
 
336
  print("\n🚀 Iniciando interfaz web...")
337
  demo.launch(