Upload folder using huggingface_hub
Browse files- 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 |
-
#
|
| 318 |
-
print("
|
| 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 |
-
|
| 326 |
-
|
| 327 |
-
|
| 328 |
-
|
| 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 |
-
|
| 347 |
-
|
| 348 |
-
|
|
|
|
| 349 |
else:
|
| 350 |
-
|
| 351 |
-
|
| 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(
|