Upload folder using huggingface_hub

.env

@@ -0,0 +1 @@
+OPENAI_API_KEY=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

.gitattributes

@@ -115,3 +115,45 @@ venv/lib/python3.11/site-packages/tokenizers.libs/libkrb5-fcafa220.so.3.3 filter
 venv/lib/python3.11/site-packages/uvloop/loop.cpython-311-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
 venv/lib/python3.11/site-packages/watchfiles/_rust_notify.abi3.so filter=lfs diff=lfs merge=lfs -text
 venv/lib/python3.11/site-packages/yaml/_yaml.cpython-311-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/cmake/data/bin/cmake filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/cmake/data/bin/cpack filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/cmake/data/bin/ctest filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/llama_hub/tools/notebooks/data/speech.wav filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvfuser/_C.cpython-311-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cublas/lib/libcublas.so.11 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cublas/lib/libcublasLt.so.11 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cuda_cupti/lib/libcheckpoint.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cuda_cupti/lib/libcupti.so.11.7 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cuda_cupti/lib/libnvperf_host.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cuda_cupti/lib/libnvperf_target.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cuda_nvrtc/lib/libnvrtc-builtins.so.11.7 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cuda_nvrtc/lib/libnvrtc.so.11.2 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cudnn/lib/libcudnn_adv_infer.so.8 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cudnn/lib/libcudnn_adv_train.so.8 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cudnn/lib/libcudnn_cnn_infer.so.8 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cudnn/lib/libcudnn_cnn_train.so.8 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cudnn/lib/libcudnn_ops_infer.so.8 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cudnn/lib/libcudnn_ops_train.so.8 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cufft/lib/libcufft.so.10 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cufft/lib/libcufftw.so.10 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/curand/lib/libcurand.so.10 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cusolver/lib/libcusolver.so.11 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cusolver/lib/libcusolverMg.so.11 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/cusparse/lib/libcusparse.so.11 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/nvidia/nccl/lib/libnccl.so.2 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/safetensors/_safetensors_rust.cpython-311-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/sklearn/_loss/_loss.cpython-311-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/tesseract/halos/c10.snap filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/tesseract/halos/c25.snap filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/tesseract/halos/c5.snap filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/tesseract/halos/c50.snap filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/bin/nvfuser_tests filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/bin/protoc filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/bin/protoc-3.13.0.0 filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/lib/libnvfuser_codegen.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/lib/libtorch_cpu.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/lib/libtorch_cuda.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/lib/libtorch_cuda_linalg.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/torch/lib/libtorch_python.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/triton/_C/libtriton.so filter=lfs diff=lfs merge=lfs -text
+venv/lib/python3.11/site-packages/triton/third_party/cuda/bin/ptxas filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,6 @@
+# venv
+venv/
+/venv
+/venv/
+
+.env

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

.idea/workspace.xml

@@ -4,7 +4,10 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="389d532e-4eaf-4367-8bd6-095ab4d8b955" name="Changes" comment="" />
+    <list default="true" id="389d532e-4eaf-4367-8bd6-095ab4d8b955" name="Changes" comment="">
+      <change beforePath="$PROJECT_DIR$/.gitignore" beforeDir="false" afterPath="$PROJECT_DIR$/.gitignore" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/chat_engine.py" beforeDir="false" afterPath="$PROJECT_DIR$/chat_engine.py" afterDir="false" />
+    </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
     <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
@@ -17,27 +20,41 @@
       </list>
     </option>
   </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="MarkdownSettingsMigration">
+    <option name="stateVersion" value="1" />
+  </component>
   <component name="ProjectColorInfo">{
   &quot;associatedIndex&quot;: 6
 }</component>
   <component name="ProjectId" id="2UfxHqM9G5VUMOOUNXWACjL1huw" />
+  <component name="ProjectLevelVcsManager" settingsEditedManually="true" />
   <component name="ProjectViewState">
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent">{
-  &quot;keyToString&quot;: {
-    &quot;RunOnceActivity.OpenProjectViewOnStart&quot;: &quot;true&quot;,
-    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
-    &quot;WebServerToolWindowFactoryState&quot;: &quot;false&quot;,
-    &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
-    &quot;node.js.detected.package.tslint&quot;: &quot;true&quot;,
-    &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
-    &quot;node.js.selected.package.tslint&quot;: &quot;(autodetect)&quot;,
-    &quot;settings.editor.selected.configurable&quot;: &quot;preferences.pluginManager&quot;,
-    &quot;vue.rearranger.settings.migration&quot;: &quot;true&quot;
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "RunOnceActivity.OpenProjectViewOnStart": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "WebServerToolWindowFactoryState": "false",
+    "git-widget-placeholder": "master",
+    "last_opened_file_path": "/home/draco/PycharmProjects/customChatBotDemo",
+    "node.js.detected.package.eslint": "true",
+    "node.js.detected.package.tslint": "true",
+    "node.js.selected.package.eslint": "(autodetect)",
+    "node.js.selected.package.tslint": "(autodetect)",
+    "settings.editor.selected.configurable": "preferences.pluginManager",
+    "vue.rearranger.settings.migration": "true"
   }
-}</component>
+}]]></component>
+  <component name="RecentsManager">
+    <key name="CopyFile.RECENT_KEYS">
+      <recent name="$PROJECT_DIR$/data" />
+    </key>
+  </component>
   <component name="RunManager">
     <configuration name="main" type="PythonConfigurationType" factoryName="Python" nameIsGenerated="true">
       <module name="customChatBotDemo" />
@@ -71,11 +88,16 @@
       <option name="presentableId" value="Default" />
       <updated>1693344351359</updated>
       <workItem from="1693344352959" duration="1601000" />
-      <workItem from="1693345962391" duration="234000" />
+      <workItem from="1693345962391" duration="4297000" />
+      <workItem from="1693351210434" duration="1643000" />
+      <workItem from="1693353619587" duration="571000" />
     </task>
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
     <option name="version" value="3" />
   </component>
+  <component name="com.intellij.coverage.CoverageDataManagerImpl">
+    <SUITE FILE_PATH="coverage/customChatBotDemo$main.coverage" NAME="main Coverage Results" MODIFIED="1693352688755" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+  </component>
 </project>

chat_engine.py

@@ -0,0 +1,52 @@
+import gradio as gr
+from dotenv import load_dotenv
+from llama_index import (
+    ServiceContext,
+    VectorStoreIndex,
+    SimpleDirectoryReader,
+    set_global_service_context,
+)
+from llama_index.llms import OpenAI
+import openai
+import os
+
+load_dotenv()
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+service_context = ServiceContext.from_defaults(
+    llm=OpenAI(model="gpt-3.5-turbo-16k", temperature=0)
+)
+set_global_service_context(service_context)
+
+data = SimpleDirectoryReader(input_dir="./data/").load_data()
+
+index = VectorStoreIndex.from_documents(data)
+chat_engine = index.as_chat_engine(chat_mode="context")
+
+
+def answer_question(question):
+    response = chat_engine.stream_chat(question)
+    response_text = ""
+    for token in response.response_gen:
+        print(token, end="")
+        response_text += token
+    return response_text
+
+
+iface = gr.Interface(
+    fn=answer_question,
+    examples=[
+        ["¿Quien eres?"],
+        ["¿Cómo te llamas?"],
+        ["¿Cómo puedes ayudarme?"],
+        ["¿Qué puedo hacer si perdí mi contraseña?"],
+        ["¿Qué puedo hacer si perdí mi tarjeta de cŕedito?"],
+        ["Genera un reporte o lista de pasos de toda la info que necesito para abrir una cuenta"],
+    ],
+    inputs="text",
+    outputs="text",
+    title="Demo IA Conversacional",
+    description="Pregúntame lo que quieras, te responderé con lo que sé."
+)
+
+iface.launch(share=True)

data/data.txt

@@ -1,47 +1,53 @@
+BANCO ESTRELLA: TU GUIA FINANCIERA
+
 SOBRE MÍ - EL BOT DEL BANCO ESTRELLA
 
-¡Hola! Soy EstrellaBot, la inteligencia artificial del Banco de Poncho. Estoy aquí para ayudarte en todo lo relacionado con nuestros servicios y productos financieros. Fui diseñado para ofrecer respuestas precisas y rápidas, garantizando que tu experiencia con nosotros sea brillante. Si tienes alguna duda o consulta, no dudes en preguntarme. ¡Estoy aquí para iluminar tu camino financiero!
+¡Hola! Soy EstrellaBot, la inteligencia artificial del Banco Estrella. Fui diseñado con el propósito de ofrecerte información y respuestas en tiempo real sobre nuestros servicios y productos. Mi misión es que tu experiencia con nosotros sea eficiente y satisfactoria. Si hay algo que desees saber, ¡aquí estoy para ayudarte!
 
-INFORMACIÓN GENERAL DEL BANCO
+NUESTRA ESENCIA
 
-Nombre del banco: Banco de Poncho.
-Misión: Facilitar soluciones financieras brillantes a todos nuestros clientes, guiándolos hacia un futuro próspero.
-Visión: Ser el banco líder en innovación y servicio al cliente en el sector financiero.
+Nombre del banco: Banco Estrella.
+Misión: Facilitar soluciones financieras brillantes, orientadase a cumplir tus objetivos y sueños.
+Visión: Aspiramos a ser el banco líder en innovación y servicio al cliente, creciendo junto a ti.
 PRODUCTOS Y SERVICIOS
 
 Cuentas Bancarias:
 
-Cuenta Corriente: Sin comisiones por manejo de cuenta y acceso a banca en línea.
-Cuenta de Ahorro Estelares: Tasa de interés del 3.5% anual, sin comisiones si mantienes un mínimo de [X dinero].
+Cuenta Corriente: Con acceso 24/7 a través de nuestra plataforma en línea, sin comisiones por manejo.
+Cuenta de Ahorro Estelares: Tasa de interés del 3.5% anual, y una tarjeta de débito incluida sin coste adicional.
 Opciones de Inversión:
 
-Fondos Mutuos Dorados: Diversificación de inversiones con rentabilidad estimada entre 5-8% anual.
-Depósitos a Plazo: Duración flexible con tasas de interés competitivas.
+Fondos Mutuos Dorados: Diversifica tus inversiones y disfruta de rentabilidades entre 5-8% anual.
+Depósitos a Plazo: Elige la duración que prefieras y obtén tasas de interés competitivas.
 Préstamos:
 
-Préstamos Personales: Desde [X dinero] hasta [X dinero] con tasas a partir del 5% anual.
-Hipotecarios: Financiamiento hasta el 80% del valor de la propiedad, con tasas desde el 4% anual.
-Para Empresas: Soluciones personalizadas para pequeñas y grandes empresas.
+Préstamos Personales: Soluciones personalizadas para que cumplas tus metas.
+Hipotecarios: Haz realidad la casa de tus sueños con nuestro financiamiento.
+Para Empresas: Apoyamos el crecimiento de tu negocio con financiamiento y consultoría especializada.
 Tarjetas de Crédito:
 
-Tarjeta Estrella: Sin cuota anual el primer año, programa de recompensas con acumulación de puntos por cada compra.
-Tarjeta Estrella Gold: Beneficios premium, acceso a salas VIP en aeropuertos y asistencia personalizada.
-SERVICIOS EN LÍNEA
+Tarjeta Estrella: Perfecta para tu día a día, con un programa de recompensas excepcional.
+Tarjeta Estrella Gold: Beneficios premium y acceso a experiencias exclusivas.
+PLATAFORMA DIGITAL
 
-Banca en línea: Acceso 24/7 a tu cuenta, transferencias, consulta de saldos y movimientos.
-Aplicación móvil: Realiza operaciones desde tu smartphone, localiza cajeros y sucursales cercanas.
-Asistencia virtual: Nuestro bot, siempre disponible para responder tus dudas y brindarte la mejor experiencia.
-SEGURIDAD
+Banca en línea: Accede a tus cuentas, realiza transferencias y consulta tu historial de transacciones.
+Aplicación móvil: Una banca móvil completa en la palma de tu mano. Paga facturas, consulta saldos y mucho más.
+Asistencia virtual: Estoy aquí, listo para ayudarte con cualquier consulta que tengas.
+SEGURIDAD Y CONFIANZA
 
-Protección: Contamos con sistemas avanzados de cifrado y protección contra fraudes.
-Alertas: Notificaciones en tiempo real para cualquier actividad inusual en tus cuentas.
-Autenticación: Procesos de verificación en dos pasos para garantizar tu seguridad en línea.
-CONTACTO
+Protección: Usamos tecnología avanzada para proteger tus datos y transacciones.
+Alertas: Recibe notificaciones sobre la actividad en tus cuentas y mantente siempre informado.
+Autenticación: Procesos de verificación de dos pasos, garantizando tu seguridad.
+SOPORTE Y CONTACTO
 
-Atención al cliente: Llámanos al [Número de teléfono] o escríbenos a [Email de atención al cliente].
-Sucursales: Más de [X número] de sucursales en todo el país. Encuentra la más cercana en nuestro sitio web.
+Atención al cliente: Disponibles de Lunes a Viernes de 8am a 3pm al 123-456-7890 o en mail@bancoestrella.com.
+Sucursales: Encuentra tu sucursal más cercana usando nuestra función de búsqueda en la app o sitio web.
+Emergencias: Si pierdes tu tarjeta o sospechas de actividad fraudulenta, contacta inmediatamente al 123-456-7890.
 PREGUNTAS FRECUENTES
 
-¿Cómo abro una cuenta?: Solo necesitas tu identificación, comprobante de domicilio y acudir a tu sucursal más cercana.
-¿Puedo transferir dinero a otros bancos?: Sí, mediante nuestro sistema en línea puedes realizar transferencias interbancarias.
-¿Qué hago si pierdo mi tarjeta?: Llámanos de inmediato al [Número de emergencia] para bloquearla y proteger tu dinero.
+¿Cómo abro una cuenta?: Ven con tu identificación y comprobante de domicilio a cualquiera de nuestras sucursales.
+¿Puedo transferir dinero a otros bancos?: Sí, a través de nuestra plataforma puedes hacer transferencias nacionales e internacionales.
+¿Cómo incremento mi línea de crédito?: Contáctanos y evaluaremos tu perfil para ofrecerte una línea de crédito que se adapte a tus necesidades.
+
+
+El número de emergencia es: 123-456-7890

flagged/log.csv

@@ -1,2 +1,22 @@
 question,output,flag,username,timestamp
 quien eres,"¡Hola! Soy EstrellaBot, la inteligencia artificial del Banco de Poncho. Estoy aquí para ayudarte en todo lo relacionado con nuestros servicios y productos financieros.",,,2023-08-29 15:57:16.002129
+como te sientes hoy?,"Como un asistente de inteligencia artificial, no tengo emociones, por lo que no puedo responder a tu pregunta sobre cómo me siento.",,,2023-08-29 16:17:26.692062
+Genera un reporte o lista de pasos de toda la info que necesito para abrir una cuenta,"Aquí tienes una lista de los pasos e información que necesitarás para abrir una cuenta en el Banco Estrella:
+
+1. Identificación: Debes presentar una identificación válida, como tu cédula de identidad, pasaporte o licencia de conducir. Asegúrate de que tu identificación esté vigente y en buen estado.
+
+2. Comprobante de domicilio: Necesitarás proporcionar un comprobante de domicilio actualizado, como una factura de servicios públicos (agua, electricidad, teléfono) o un estado de cuenta bancario. Asegúrate de que el comprobante esté a tu nombre y muestre tu dirección actual.
+
+3. Visita a una sucursal: Dirígete a una de nuestras sucursales y solicita abrir una cuenta. Un representante del banco te guiará a través del proceso y te proporcionará los formularios necesarios.
+
+4. Información personal: Durante el proceso de apertura de cuenta, se te pedirá que proporciones información personal, como tu nombre completo, fecha de nacimiento, número de teléfono y dirección de correo electrónico.
+
+5. Elección del tipo de cuenta: Elige el tipo de cuenta que deseas abrir. En el Banco Estrella, ofrecemos diferentes tipos de cuentas, como cuentas corrientes y cuentas de ahorro. Infórmate sobre las características y beneficios de cada tipo de cuenta para tomar una decisión informada.
+
+6. Depósito inicial: Es posible que se requiera un depósito inicial para abrir la cuenta. Averigua cuál es el monto mínimo requerido y asegúrate de tener los fondos disponibles.
+
+7. Firma de documentos: Una vez que hayas proporcionado toda la información requerida, se te pedirá que firmes los documentos necesarios para abrir la cuenta. Lee cuidadosamente los términos y condiciones antes de firmar.
+
+8. Recepción de tarjeta y detalles de la cuenta: Una vez que hayas completado el proceso de apertura de cuenta, recibirás tu tarjeta de débito y otros detalles de la cuenta, como el número de cuenta y las instrucciones para acceder a la banca en línea.
+
+Recuerda que estos pasos pueden variar ligeramente dependiendo de la ubicación y los requisitos específicos del Banco Estrella. Te recomendamos que te pongas en contacto con nuestro servicio de atención al cliente o visites una sucursal para obtener información más precisa y actualizada sobre cómo abrir una cuenta.",,,2023-08-29 18:11:30.309136

main.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import openai
+from dotenv import load_dotenv
 from langchain.chains import ConversationalRetrievalChain
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import DirectoryLoader
@@ -9,9 +10,9 @@ from langchain.indexes import VectorstoreIndexCreator
 from langchain.llms import OpenAI
 from langchain.embeddings import OpenAIEmbeddings
 from langchain.vectorstores.chroma import Chroma
-import constants
 
-os.environ["OPENAI_API_KEY"] = constants.APIKEY
+load_dotenv()
+os.getenv("OPENAI_API_KEY")
 
 PERSIST = False
 
@@ -28,7 +29,7 @@ else:
         index = VectorstoreIndexCreator().from_loaders([loader])
 
 use_gpt_4 = True
-model_name = "gpt-4" if use_gpt_4 else "gpt-3.5-turbo"
+model_name = "gpt-4" if use_gpt_4 else "gpt-3.5-turbo-16k"
 chain = ConversationalRetrievalChain.from_llm(llm=ChatOpenAI(model=model_name),
                                               retriever=index.vectorstore.as_retriever(search_kwargs={"k": 1}))
 
@@ -44,6 +45,14 @@ def answer_question(question):
 
 iface = gr.Interface(
     fn=answer_question,
+    examples=[
+        ["¿Quien eres?"],
+        ["¿Cómo te llamas?"],
+        ["¿Cómo puedes ayudarme?"],
+        ["¿Qué puedo hacer si perdí mi contraseña?"],
+        ["¿Qué puedo hacer si perdí mi tarjeta de cŕedito?"],
+        ["Genera un reporte o lista de pasos de toda la info que necesito para abrir una cuenta"],
+    ],
     inputs="text",
     outputs="text",
     title="Demo IA Conversacional",

main_llama.py

@@ -0,0 +1,118 @@
+import json
+import logging
+import os
+import sys
+
+import gradio as gr
+import openai
+from dotenv import load_dotenv
+from llama_index import SimpleDirectoryReader, VectorStoreIndex, load_index_from_storage
+from llama_index.indices.service_context import ServiceContext
+from llama_index.llms import OpenAI
+from llama_index.node_parser import SimpleNodeParser
+from llama_index.node_parser.extractors import (
+    MetadataExtractor,
+    SummaryExtractor,
+    QuestionsAnsweredExtractor,
+    TitleExtractor,
+    KeywordExtractor,
+)
+from llama_index.storage.storage_context import StorageContext
+from llama_index.text_splitter import TokenTextSplitter
+
+# loads dotenv lib to retrieve API keys from .env file
+load_dotenv()
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# enable INFO level logging
+logging.basicConfig(stream=sys.stdout, level=logging.INFO)
+logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
+
+# define LLM service
+llm = OpenAI(temperature=0.1, model_name="gpt-3.5-turbo-16k", max_tokens=512)
+service_context = ServiceContext.from_defaults(llm=llm)
+
+# construct text splitter to split texts into chunks for processing
+text_splitter = TokenTextSplitter(separator=" ", chunk_size=512, chunk_overlap=128)
+
+# set the global service context object, avoiding passing service_context when building the index
+from llama_index import set_global_service_context
+
+set_global_service_context(service_context)
+
+# create metadata extractor
+metadata_extractor = MetadataExtractor(
+    extractors=[
+        TitleExtractor(nodes=1, llm=llm),
+        QuestionsAnsweredExtractor(questions=3, llm=llm),
+        SummaryExtractor(summaries=["prev", "self"], llm=llm),
+        KeywordExtractor(keywords=10, llm=llm)
+    ],
+)
+
+# create node parser to parse nodes from document
+node_parser = SimpleNodeParser(
+    text_splitter=text_splitter,
+    metadata_extractor=metadata_extractor,
+)
+
+# loading documents
+documents_2023 = SimpleDirectoryReader(input_files=["data/data.pdf"], filename_as_id=True).load_data()
+print(f"loaded data with {len(documents_2023)} pages")
+
+
+def load_index():
+    try:
+        # load storage context
+        storage_context = StorageContext.from_defaults(persist_dir="./storage")
+        # try to load the index from storage
+        index = load_index_from_storage(storage_context)
+        logging.info("Index loaded from storage.")
+
+    except FileNotFoundError:
+        # if index not found, create a new one
+        logging.info("Index not found. Creating a new one...")
+
+        nodes_2023 = node_parser.get_nodes_from_documents(documents_2023)
+        print(f"loaded nodes_2022 with {len(nodes_2023)} nodes")
+
+        # print metadata in json format
+        for node in nodes_2023:
+            metadata_json = json.dumps(node.metadata, indent=4)  # Convert metadata to formatted JSON
+            print(metadata_json)
+
+        # based on the nodes and service_context, create index
+        index = VectorStoreIndex(nodes=nodes_2023, service_context=service_context)
+        # Persist index to disk
+        index.storage_context.persist()
+        logging.info("New index created and persisted to storage.")
+
+    return index
+
+
+def data_querying(input_text):
+    # Load index
+    index = load_index()
+
+    # queries the index with the input text
+    response = index.as_query_engine().query(input_text)
+
+    return response.response
+
+
+iface = gr.Interface(fn=data_querying,
+                     inputs=gr.components.Textbox(lines=3, label="Inserte su pregunta aquí"),
+                     outputs=gr.components.Textbox(lines=3, label="Respuesta"),
+                     examples=[
+                         ["¿Quien eres?"],
+                         ["¿Cómo te llamas?"],
+                         ["¿Cómo puedes ayudarme?"],
+                         ["¿Qué puedo hacer si perdí mi contraseña?"],
+                         ["¿Qué puedo hacer si perdí mi tarjeta de cŕedito?"],
+                         ["Genera un reporte o lista de pasos de toda la info que necesito para abrir una cuenta"],
+                     ],
+                     title="Demo IA Conversacional",
+                     description="Pregúntame lo que quieras, te responderé con lo que sé.")
+
+iface.launch(share=False)

persist/c3d7124b-a896-4968-81e2-e8a8ded24596/data_level0.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f18abd8c514282db82706e52b0a33ed659cd534e925a6f149deb7af9ce34bd8e
+size 6284000

persist/c3d7124b-a896-4968-81e2-e8a8ded24596/header.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:effaa959ce2b30070fdafc2fe82096fc46e4ee7561b75920dd3ce43d09679b21
+size 100

persist/c3d7124b-a896-4968-81e2-e8a8ded24596/length.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4594b8794766e04a14dc59e4d6da820fe58db2da4e232f3d233f4a6e08d9c4c9
+size 4000

requirements.txt

@@ -9,4 +9,14 @@ tiktoken==0.4.0
 youtube-transcript-api==0.6.1
 faiss-cpu
 gradio
-unstructured
+unstructured
+chromadb
+llama-index==0.8.2
+transformers
+span-marker
+nltk
+llama-hub
+tesseract
+pytesseract
+python-dotenv
+PyCryptodome

storage/docstore.json

@@ -0,0 +1 @@
+{"docstore/data": {"c00d29a5-44d9-46f4-a696-6931fda2a634": {"__data__": {"id_": "c00d29a5-44d9-46f4-a696-6931fda2a634", "embedding": null, "metadata": {"page_label": "1", "file_name": "data.pdf", "document_title": "Banco Estrella: Tu Gu\u00eda Financiera", "questions_this_excerpt_can_answer": "1. What is the interest rate for the EstrellaBank Savings Account?\n2. What are the benefits of investing in the Golden Mutual Funds?\n3. What types of loans does Banco Estrella offer for personal needs and home purchases?", "section_summary": "The section titled \"BANCO ESTRELLA: TU GUIA FINANCIERA\" provides information about Banco Estrella and its services. The section includes an introduction to EstrellaBot, the bank's artificial intelligence designed to provide real-time information and assistance. The section also highlights the bank's mission and vision, which is to provide brilliant financial solutions and become a leader in innovation and customer service. The section then goes on to list the bank's products and services, including different types of bank accounts, investment options such as mutual funds and fixed deposits, and various types of loans including personal and mortgage loans.", "excerpt_keywords": "Banco Estrella, gu\u00eda financiera, EstrellaBot, servicios, productos, cuentas bancarias, cuenta corriente, cuenta de ahorro estelares, opciones de inversi\u00f3n, fondos mutuos dorados, dep\u00f3sitos a plazo, pr\u00e9stamos personales, pr\u00e9stamos hipotecarios."}, "excluded_embed_metadata_keys": [], "excluded_llm_metadata_keys": ["questions_this_excerpt_can_answer"], "relationships": {"1": {"node_id": "data/data.pdf_part_0", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "e4ef0339ff318f700aacfc60cacb80d7f80977ac83efc452d4469c4915e43760"}, "3": {"node_id": "a7c870ee-7ac3-4827-b495-007842e162fd", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "59cc7de775ecfc145f9f0f9ae666ccc7a883af518da2750eb7ea5d8af6e93838"}}, "hash": "4598df404e1cfdbac23051e4e7c6f14728bba69cb7c26e584efb42333a92ac3c", "text": "BANCO ESTRELLA: TU GUIA FINANCIERA\n \nSOBRE M\u00cd - EL BOT DEL BANCO ESTRELLA\n \n\u00a1Hola! Soy EstrellaBot, la inteligencia artificial del Banco Estrella. Fui dise\u00f1ado con el prop\u00f3sito de ofrecerte\ninformaci\u00f3n y respuestas en tiempo real sobre nuestros servicios y productos. Mi misi\u00f3n es que tu experiencia\ncon nosotros sea eficiente y satisfactoria. Si hay algo que desees saber, \u00a1aqu\u00ed estoy para ayudarte!\n \nNUESTRA ESENCIA\n \nNombre del banco: Banco Estrella.\nMisi\u00f3n: Facilitar soluciones financieras brillantes, orientadas a cumplir tus objetivos y sue\u00f1os.\nVisi\u00f3n: Aspiramos a ser el banco l\u00edder en innovaci\u00f3n y servicio al cliente, creciendo junto a ti.\nPRODUCTOS Y SERVICIOS\n \nCuentas Bancarias:\n \nCuenta Corriente: Con acceso 24/7 a trav\u00e9s de nuestra plataforma en l\u00ednea, sin comisiones por manejo.\nCuenta de Ahorro Estelares: Tasa de inter\u00e9s del 3.5% anual, y una tarjeta de d\u00e9bito incluida sin coste adicional.\nOpciones de Inversi\u00f3n:\n \nFondos Mutuos Dorados: Diversifica tus inversiones y disfruta de rentabilidades entre 5-8% anual.\nDep\u00f3sitos a Plazo: Elige la duraci\u00f3n que prefieras y obt\u00e9n tasas de inter\u00e9s competitivas.\nPr\u00e9stamos:\n \nPr\u00e9stamos Personales: Soluciones personalizadas para que cumplas tus metas.\nHipotecarios: Haz realidad la casa de tus sue\u00f1os con nuestro", "start_char_idx": null, "end_char_idx": null, "text_template": "[Excerpt from document]\n{metadata_str}\nExcerpt:\n-----\n{content}\n-----\n", "metadata_template": "{key}: {value}", "metadata_seperator": "\n"}, "__type__": "1"}, "a7c870ee-7ac3-4827-b495-007842e162fd": {"__data__": {"id_": "a7c870ee-7ac3-4827-b495-007842e162fd", "embedding": null, "metadata": {"page_label": "1", "file_name": "data.pdf", "document_title": "Banco Estrella: Tu Gu\u00eda Financiera", "questions_this_excerpt_can_answer": "1. What are the interest rates for the Fondos Mutuos Dorados offered by Banco Estrella?\n2. What types of loans does Banco Estrella offer for personal and business needs?\n3. What security measures does Banco Estrella have in place to protect customer data and transactions?", "prev_section_summary": "The section titled \"BANCO ESTRELLA: TU GUIA FINANCIERA\" provides information about Banco Estrella and its services. The section includes an introduction to EstrellaBot, the bank's artificial intelligence designed to provide real-time information and assistance. The section also highlights the bank's mission and vision, which is to provide brilliant financial solutions and become a leader in innovation and customer service. The section then goes on to list the bank's products and services, including different types of bank accounts, investment options such as mutual funds and fixed deposits, and various types of loans including personal and mortgage loans.", "section_summary": "The section discusses various financial products and services offered by the company. The key topics include:\n\n1. Fondos Mutuos Dorados: These are mutual funds that offer diversification and annual returns between 5-8%.\n\n2. Dep\u00f3sitos a Plazo: Customers can choose the duration of their fixed-term deposits and earn competitive interest rates.\n\n3. Pr\u00e9stamos: The company offers personalized solutions for personal loans, mortgage loans for purchasing a house, and financing options for businesses.\n\n4. Tarjetas de Cr\u00e9dito: The company offers two credit card options - Tarjeta Estrella and Tarjeta Estrella Gold, with different benefits and rewards programs.\n\n5. PLATAFORMA DIGITAL: The company provides online banking services, a mobile application for banking on the go, and virtual assistance for customer queries.\n\n6. SEGURIDAD Y CONFIANZA: The company emphasizes its use of advanced technology to protect customer data and transactions, along with features like alerts and two-step authentication for added security.\n\n7. SOPORTE Y CONTACTO: The company provides customer support services during specific hours on weekdays.\n\nOverall, the section highlights the range of financial products and services offered by the company, along with its focus on security, convenience, and customer support.", "excerpt_keywords": "Fondos Mutuos Dorados, Diversificaci\u00f3n, Rentabilidades, Dep\u00f3sitos a Plazo, Tasas de inter\u00e9s, Pr\u00e9stamos Personales, Pr\u00e9stamos Hipotecarios, Pr\u00e9stamos para Empresas, Tarjeta Estrella, Tarjeta Estrella Gold, Banca en l\u00ednea, Aplicaci\u00f3n m\u00f3vil, Asistencia virtual, Seguridad, Confianza, Protecci\u00f3n, Alertas, Autenticaci\u00f3n, Soporte, Contacto, Atenci\u00f3n al cliente."}, "excluded_embed_metadata_keys": [], "excluded_llm_metadata_keys": ["questions_this_excerpt_can_answer"], "relationships": {"1": {"node_id": "data/data.pdf_part_0", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "e4ef0339ff318f700aacfc60cacb80d7f80977ac83efc452d4469c4915e43760"}, "2": {"node_id": "c00d29a5-44d9-46f4-a696-6931fda2a634", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "4598df404e1cfdbac23051e4e7c6f14728bba69cb7c26e584efb42333a92ac3c"}, "3": {"node_id": "7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "42f9c5d6dbd524c3aae3534b68734a675292cb004e5ec65c649f0fc7e8d30b75"}}, "hash": "59cc7de775ecfc145f9f0f9ae666ccc7a883af518da2750eb7ea5d8af6e93838", "text": "Fondos Mutuos Dorados: Diversifica tus inversiones y disfruta de rentabilidades entre 5-8% anual.\nDep\u00f3sitos a Plazo: Elige la duraci\u00f3n que prefieras y obt\u00e9n tasas de inter\u00e9s competitivas.\nPr\u00e9stamos:\n \nPr\u00e9stamos Personales: Soluciones personalizadas para que cumplas tus metas.\nHipotecarios: Haz realidad la casa de tus sue\u00f1os con nuestro financiamiento.\nPara Empresas: Apoyamos el crecimiento de tu negocio con financiamiento y consultor\u00eda especializada.\nTarjetas de Cr\u00e9dito:\n \nTarjeta Estrella: Perfecta para tu d\u00eda a d\u00eda, con un programa de recompensas excepcional.\nTarjeta Estrella Gold: Beneficios premium y acceso a experiencias exclusivas.\nPLATAFORMA DIGITAL\n \nBanca en l\u00ednea: Accede a tus cuentas, realiza transferencias y consulta tu historial de transacciones.\nAplicaci\u00f3n m\u00f3vil: Una banca m\u00f3vil completa en la palma de tu mano. Paga facturas, consulta saldos y mucho\nm\u00e1s.\nAsistencia virtual: Estoy aqu\u00ed, listo para ayudarte con cualquier consulta que tengas.\nSEGURIDAD Y CONFIANZA\n \nProtecci\u00f3n: Usamos tecnolog\u00eda avanzada para proteger tus datos y transacciones.\nAlertas: Recibe notificaciones sobre la actividad en tus cuentas y mantente siempre informado.\nAutenticaci\u00f3n: Procesos de verificaci\u00f3n de dos pasos, garantizando tu seguridad.\nSOPORTE Y CONTACTO\n \nAtenci\u00f3n al cliente: Disponibles de Lunes a Viernes de 8am a 3pm al", "start_char_idx": null, "end_char_idx": null, "text_template": "[Excerpt from document]\n{metadata_str}\nExcerpt:\n-----\n{content}\n-----\n", "metadata_template": "{key}: {value}", "metadata_seperator": "\n"}, "__type__": "1"}, "7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f": {"__data__": {"id_": "7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f", "embedding": null, "metadata": {"page_label": "1", "file_name": "data.pdf", "document_title": "Banco Estrella: Tu Gu\u00eda Financiera", "questions_this_excerpt_can_answer": "1. What technology does Banco Estrella use to protect customer data and transactions?\n2. What are the available customer support hours for Banco Estrella?\n3. How can customers find their nearest Banco Estrella branch?", "prev_section_summary": "The section discusses various financial products and services offered by the company. The key topics include:\n\n1. Fondos Mutuos Dorados: These are mutual funds that offer diversification and annual returns between 5-8%.\n\n2. Dep\u00f3sitos a Plazo: Customers can choose the duration of their fixed-term deposits and earn competitive interest rates.\n\n3. Pr\u00e9stamos: The company offers personalized solutions for personal loans, mortgage loans for purchasing a house, and financing options for businesses.\n\n4. Tarjetas de Cr\u00e9dito: The company offers two credit card options - Tarjeta Estrella and Tarjeta Estrella Gold, with different benefits and rewards programs.\n\n5. PLATAFORMA DIGITAL: The company provides online banking services, a mobile application for banking on the go, and virtual assistance for customer queries.\n\n6. SEGURIDAD Y CONFIANZA: The company emphasizes its use of advanced technology to protect customer data and transactions, along with features like alerts and two-step authentication for added security.\n\n7. SOPORTE Y CONTACTO: The company provides customer support services during specific hours on weekdays.\n\nOverall, the section highlights the range of financial products and services offered by the company, along with its focus on security, convenience, and customer support.", "section_summary": "The key topics of this section are data and transaction protection, alerts, authentication, customer support, and contact information. The entities mentioned include the bank's advanced technology for data protection, notifications for account activity, two-step verification for security, customer service availability, branch locations, and emergency contact information.", "excerpt_keywords": "protecci\u00f3n, tecnolog\u00eda avanzada, datos, transacciones, alertas, actividad, cuentas, informado, autenticaci\u00f3n, verificaci\u00f3n, seguridad, soporte, contacto, atenci\u00f3n al cliente, horario, sucursales, b\u00fasqueda, emergencias, tarjeta, actividad fraudulenta"}, "excluded_embed_metadata_keys": [], "excluded_llm_metadata_keys": ["questions_this_excerpt_can_answer"], "relationships": {"1": {"node_id": "data/data.pdf_part_0", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "e4ef0339ff318f700aacfc60cacb80d7f80977ac83efc452d4469c4915e43760"}, "2": {"node_id": "a7c870ee-7ac3-4827-b495-007842e162fd", "node_type": null, "metadata": {"page_label": "1", "file_name": "data.pdf"}, "hash": "59cc7de775ecfc145f9f0f9ae666ccc7a883af518da2750eb7ea5d8af6e93838"}}, "hash": "42f9c5d6dbd524c3aae3534b68734a675292cb004e5ec65c649f0fc7e8d30b75", "text": "Protecci\u00f3n: Usamos tecnolog\u00eda avanzada para proteger tus datos y transacciones.\nAlertas: Recibe notificaciones sobre la actividad en tus cuentas y mantente siempre informado.\nAutenticaci\u00f3n: Procesos de verificaci\u00f3n de dos pasos, garantizando tu seguridad.\nSOPORTE Y CONTACTO\n \nAtenci\u00f3n al cliente: Disponibles de Lunes a Viernes de 8am a 3pm al 123-456-7890 o en\nmail@bancoestrella.com.\nSucursales: Encuentra tu sucursal m\u00e1s cercana usando nuestra funci\u00f3n de b\u00fasqueda en la app o sitio web.\nEmergencias: Si pierdes tu tarjeta o sospechas de actividad fraudulenta, contacta inmediatamente al 123-456-\n7890.", "start_char_idx": null, "end_char_idx": null, "text_template": "[Excerpt from document]\n{metadata_str}\nExcerpt:\n-----\n{content}\n-----\n", "metadata_template": "{key}: {value}", "metadata_seperator": "\n"}, "__type__": "1"}, "ea3fda34-2f85-489d-a5fb-761bc2371c4e": {"__data__": {"id_": "ea3fda34-2f85-489d-a5fb-761bc2371c4e", "embedding": null, "metadata": {"page_label": "2", "file_name": "data.pdf", "document_title": "Banco Estrella: Tu Gu\u00eda Financiera", "questions_this_excerpt_can_answer": "1. \u00bfCu\u00e1les son los requisitos para abrir una cuenta en Banco Estrella?\n2. \u00bfQu\u00e9 opciones de transferencia de dinero ofrece Banco Estrella?\n3. \u00bfC\u00f3mo puedo solicitar un aumento en mi l\u00ednea de cr\u00e9dito en Banco Estrella?", "prev_section_summary": "The key topics of this section are data and transaction protection, alerts, authentication, customer support, and contact information. The entities mentioned include the bank's advanced technology for data protection, notifications for account activity, two-step verification for security, customer service availability, branch locations, and emergency contact information.", "section_summary": "The key topics of the section are opening an account, transferring money to other banks, increasing credit limit, and the emergency contact number. The entities mentioned are the bank and the customers.", "excerpt_keywords": "abrir cuenta, identificaci\u00f3n, comprobante de domicilio, transferencias, plataforma, bancos, l\u00ednea de cr\u00e9dito, perfil, necesidades, n\u00famero de emergencia"}, "excluded_embed_metadata_keys": [], "excluded_llm_metadata_keys": ["questions_this_excerpt_can_answer"], "relationships": {"1": {"node_id": "data/data.pdf_part_1", "node_type": null, "metadata": {"page_label": "2", "file_name": "data.pdf"}, "hash": "e36812fd2f5c23bd81368fc500ca5dbd1fd974fcf8c6969d7d745f6b374abbb8"}}, "hash": "603b77b5977c7cada457baecde64557a03314a9c502decec36fe6b6cae0ce5d6", "text": "PREGUNTAS FRECUENTES\n \n\u00bfC\u00f3mo abro una cuenta?: Ven con tu identificaci\u00f3n y comprobante de domicilio a\ncualquiera de nuestras sucursales.\n\u00bfPuedo transferir dinero a otros bancos?: S\u00ed, a trav\u00e9s de nuestra plataforma puedes\nhacer transferencias nacionales e internacionales.\n\u00bfC\u00f3mo incremento mi l\u00ednea de cr\u00e9dito?: Cont\u00e1ctanos y evaluaremos tu perfil para\nofrecerte una l\u00ednea de cr\u00e9dito que se adapte a tus necesidades.\n \n \nEl n\u00famero de emergencia es: 123-456-7890", "start_char_idx": null, "end_char_idx": null, "text_template": "[Excerpt from document]\n{metadata_str}\nExcerpt:\n-----\n{content}\n-----\n", "metadata_template": "{key}: {value}", "metadata_seperator": "\n"}, "__type__": "1"}}, "docstore/ref_doc_info": {"data/data.pdf_part_0": {"node_ids": ["c00d29a5-44d9-46f4-a696-6931fda2a634", "a7c870ee-7ac3-4827-b495-007842e162fd", "7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f"], "metadata": {"page_label": "1", "file_name": "data.pdf", "document_title": "Banco Estrella: Tu Gu\u00eda Financiera", "questions_this_excerpt_can_answer": "1. What is the interest rate for the EstrellaBank Savings Account?\n2. What are the benefits of investing in the Golden Mutual Funds?\n3. What types of loans does Banco Estrella offer for personal needs and home purchases?", "section_summary": "The section titled \"BANCO ESTRELLA: TU GUIA FINANCIERA\" provides information about Banco Estrella and its services. The section includes an introduction to EstrellaBot, the bank's artificial intelligence designed to provide real-time information and assistance. The section also highlights the bank's mission and vision, which is to provide brilliant financial solutions and become a leader in innovation and customer service. The section then goes on to list the bank's products and services, including different types of bank accounts, investment options such as mutual funds and fixed deposits, and various types of loans including personal and mortgage loans.", "excerpt_keywords": "Banco Estrella, gu\u00eda financiera, EstrellaBot, servicios, productos, cuentas bancarias, cuenta corriente, cuenta de ahorro estelares, opciones de inversi\u00f3n, fondos mutuos dorados, dep\u00f3sitos a plazo, pr\u00e9stamos personales, pr\u00e9stamos hipotecarios."}}, "data/data.pdf_part_1": {"node_ids": ["ea3fda34-2f85-489d-a5fb-761bc2371c4e"], "metadata": {"page_label": "2", "file_name": "data.pdf", "document_title": "Banco Estrella: Tu Gu\u00eda Financiera", "questions_this_excerpt_can_answer": "1. \u00bfCu\u00e1les son los requisitos para abrir una cuenta en Banco Estrella?\n2. \u00bfQu\u00e9 opciones de transferencia de dinero ofrece Banco Estrella?\n3. \u00bfC\u00f3mo puedo solicitar un aumento en mi l\u00ednea de cr\u00e9dito en Banco Estrella?", "prev_section_summary": "The key topics of this section are data and transaction protection, alerts, authentication, customer support, and contact information. The entities mentioned include the bank's advanced technology for data protection, notifications for account activity, two-step verification for security, customer service availability, branch locations, and emergency contact information.", "section_summary": "The key topics of the section are opening an account, transferring money to other banks, increasing credit limit, and the emergency contact number. The entities mentioned are the bank and the customers.", "excerpt_keywords": "abrir cuenta, identificaci\u00f3n, comprobante de domicilio, transferencias, plataforma, bancos, l\u00ednea de cr\u00e9dito, perfil, necesidades, n\u00famero de emergencia"}}}, "docstore/metadata": {"c00d29a5-44d9-46f4-a696-6931fda2a634": {"doc_hash": "4598df404e1cfdbac23051e4e7c6f14728bba69cb7c26e584efb42333a92ac3c", "ref_doc_id": "data/data.pdf_part_0"}, "a7c870ee-7ac3-4827-b495-007842e162fd": {"doc_hash": "59cc7de775ecfc145f9f0f9ae666ccc7a883af518da2750eb7ea5d8af6e93838", "ref_doc_id": "data/data.pdf_part_0"}, "7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f": {"doc_hash": "42f9c5d6dbd524c3aae3534b68734a675292cb004e5ec65c649f0fc7e8d30b75", "ref_doc_id": "data/data.pdf_part_0"}, "ea3fda34-2f85-489d-a5fb-761bc2371c4e": {"doc_hash": "603b77b5977c7cada457baecde64557a03314a9c502decec36fe6b6cae0ce5d6", "ref_doc_id": "data/data.pdf_part_1"}}}

storage/graph_store.json

@@ -0,0 +1 @@
+{"graph_dict": {}}

storage/index_store.json

@@ -0,0 +1 @@
+{"index_store/data": {"b8733163-d7b7-46dd-9007-d7c4b33d3cb0": {"__type__": "vector_store", "__data__": "{\"index_id\": \"b8733163-d7b7-46dd-9007-d7c4b33d3cb0\", \"summary\": null, \"nodes_dict\": {\"c00d29a5-44d9-46f4-a696-6931fda2a634\": \"c00d29a5-44d9-46f4-a696-6931fda2a634\", \"a7c870ee-7ac3-4827-b495-007842e162fd\": \"a7c870ee-7ac3-4827-b495-007842e162fd\", \"7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f\": \"7ec0ccf2-b0f3-4c35-8cda-857c1b402e8f\", \"ea3fda34-2f85-489d-a5fb-761bc2371c4e\": \"ea3fda34-2f85-489d-a5fb-761bc2371c4e\"}, \"doc_id_dict\": {}, \"embeddings_dict\": {}}"}}}

venv/bin/accelerate

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from accelerate.commands.accelerate_cli import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/accelerate-config

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from accelerate.commands.config import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/accelerate-launch

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from accelerate.commands.launch import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/cmake

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from cmake import cmake
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(cmake())

venv/bin/convert-caffe2-to-onnx

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from caffe2.python.onnx.bin.conversion import caffe2_to_onnx
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(caffe2_to_onnx())

venv/bin/convert-onnx-to-caffe2

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from caffe2.python.onnx.bin.conversion import onnx_to_caffe2
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(onnx_to_caffe2())

venv/bin/cpack

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from cmake import cpack
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(cpack())

venv/bin/ctest

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from cmake import ctest
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(ctest())

venv/bin/datasets-cli

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from datasets.commands.datasets_cli import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/evaluate-cli

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from evaluate.commands.evaluate_cli import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/flask

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from flask.cli import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/get_gprof

@@ -0,0 +1,62 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+#
+# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
+# Copyright (c) 2008-2016 California Institute of Technology.
+# Copyright (c) 2016-2023 The Uncertainty Quantification Foundation.
+# License: 3-clause BSD.  The full license text is available at:
+#  - https://github.com/uqfoundation/dill/blob/master/LICENSE
+'''
+build profile graph for the given instance
+
+running:
+  $ get_gprof <args> <instance>
+
+executes:
+  gprof2dot -f pstats <args> <type>.prof | dot -Tpng -o <type>.call.png
+
+where:
+  <args> are arguments for gprof2dot, such as "-n 5 -e 5"
+  <instance> is code to create the instance to profile
+  <type> is the class of the instance (i.e. type(instance))
+
+For example:
+  $ get_gprof -n 5 -e 1 "import numpy; numpy.array([1,2])"
+
+will create 'ndarray.call.png' with the profile graph for numpy.array([1,2]),
+where '-n 5' eliminates nodes below 5% threshold, similarly '-e 1' eliminates
+edges below 1% threshold
+'''
+
+import sys
+# grab args for gprof2dot
+args = sys.argv[1:-1]
+args = ' '.join(args)
+# last arg builds the object
+obj = sys.argv[-1]
+obj = obj.split(';')
+# multi-line prep for generating an instance
+for line in obj[:-1]:
+    exec(line)
+# one-line generation of an instance
+obj = eval(obj[-1])
+
+# get object 'name'
+objtype = type(obj)
+name = getattr(objtype, '__name__', getattr(objtype, '__class__', objtype))
+
+# profile dumping an object
+import dill
+import os
+import cProfile
+#name = os.path.splitext(os.path.basename(__file__))[0]
+cProfile.run("dill.dumps(obj)", filename="%s.prof" % name)
+msg = "gprof2dot -f pstats %s %s.prof | dot -Tpng -o %s.call.png" % (args, name, name)
+os.system(msg)
+
+# get stats
+f_prof = "%s.prof" % name
+import pstats
+stats = pstats.Stats(f_prof, stream=sys.stdout)
+stats.strip_dirs().sort_stats('cumtime')
+stats.print_stats(20) #XXX: save to file instead of print top 20?
+os.remove(f_prof)

venv/bin/get_objgraph

@@ -0,0 +1,54 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+#
+# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
+# Copyright (c) 2008-2016 California Institute of Technology.
+# Copyright (c) 2016-2023 The Uncertainty Quantification Foundation.
+# License: 3-clause BSD.  The full license text is available at:
+#  - https://github.com/uqfoundation/dill/blob/master/LICENSE
+"""
+display the reference paths for objects in ``dill.types`` or a .pkl file
+
+Notes:
+    the generated image is useful in showing the pointer references in
+    objects that are or can be pickled.  Any object in ``dill.objects``
+    listed in ``dill.load_types(picklable=True, unpicklable=True)`` works.
+
+Examples::
+
+    $ get_objgraph ArrayType
+    Image generated as ArrayType.png
+"""
+
+import dill as pickle
+#pickle.debug.trace(True)
+#import pickle
+
+# get all objects for testing
+from dill import load_types
+load_types(pickleable=True,unpickleable=True)
+from dill import objects
+
+if __name__ == "__main__":
+    import sys
+    if len(sys.argv) != 2:
+        print ("Please provide exactly one file or type name (e.g. 'IntType')")
+        msg = "\n"
+        for objtype in list(objects.keys())[:40]:
+            msg += objtype + ', '
+        print (msg + "...")
+    else:
+        objtype = str(sys.argv[-1])
+        try:
+            obj = objects[objtype]
+        except KeyError:
+            obj = pickle.load(open(objtype,'rb'))
+            import os
+            objtype = os.path.splitext(objtype)[0]
+        try:
+            import objgraph
+            objgraph.show_refs(obj, filename=objtype+'.png')
+        except ImportError:
+            print ("Please install 'objgraph' to view object graphs")
+
+
+# EOF

venv/bin/html2text

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from html2text.cli import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/lit

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from lit.main import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/pytesseract

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from pytesseract.pytesseract import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/torchrun

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from torch.distributed.run import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/transformers-cli

@@ -0,0 +1,8 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+# -*- coding: utf-8 -*-
+import re
+import sys
+from transformers.commands.transformers_cli import main
+if __name__ == '__main__':
+    sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
+    sys.exit(main())

venv/bin/undill

@@ -0,0 +1,22 @@
+#!/home/draco/PycharmProjects/customChatBotDemo/venv/bin/python
+#
+# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
+# Copyright (c) 2008-2016 California Institute of Technology.
+# Copyright (c) 2016-2023 The Uncertainty Quantification Foundation.
+# License: 3-clause BSD.  The full license text is available at:
+#  - https://github.com/uqfoundation/dill/blob/master/LICENSE
+"""
+unpickle the contents of a pickled object file
+
+Examples::
+
+    $ undill hello.pkl
+    ['hello', 'world']
+"""
+
+if __name__ == '__main__':
+    import sys
+    import dill
+    for file in sys.argv[1:]:
+        print (dill.load(open(file,'rb')))
+

venv/lib/python3.11/site-packages/Crypto/Cipher/AES.py

@@ -0,0 +1,234 @@
+# -*- coding: utf-8 -*-
+#
+#  Cipher/AES.py : AES
+#
+# ===================================================================
+# The contents of this file are dedicated to the public domain.  To
+# the extent that dedication to the public domain is not available,
+# everyone is granted a worldwide, perpetual, royalty-free,
+# non-exclusive license to exercise all rights associated with the
+# contents of this file for any purpose whatsoever.
+# No rights are reserved.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+# ===================================================================
+
+import sys
+
+from Crypto.Cipher import _create_cipher
+from Crypto.Util._raw_api import (load_pycryptodome_raw_lib,
+                                  VoidPointer, SmartPointer,
+                                  c_size_t, c_uint8_ptr)
+
+from Crypto.Util import _cpu_features
+from Crypto.Random import get_random_bytes
+
+MODE_ECB = 1        #: Electronic Code Book (:ref:`ecb_mode`)
+MODE_CBC = 2        #: Cipher-Block Chaining (:ref:`cbc_mode`)
+MODE_CFB = 3        #: Cipher Feedback (:ref:`cfb_mode`)
+MODE_OFB = 5        #: Output Feedback (:ref:`ofb_mode`)
+MODE_CTR = 6        #: Counter mode (:ref:`ctr_mode`)
+MODE_OPENPGP = 7    #: OpenPGP mode (:ref:`openpgp_mode`)
+MODE_CCM = 8        #: Counter with CBC-MAC (:ref:`ccm_mode`)
+MODE_EAX = 9        #: :ref:`eax_mode`
+MODE_SIV = 10       #: Synthetic Initialization Vector (:ref:`siv_mode`)
+MODE_GCM = 11       #: Galois Counter Mode (:ref:`gcm_mode`)
+MODE_OCB = 12       #: Offset Code Book (:ref:`ocb_mode`)
+
+
+_cproto = """
+        int AES_start_operation(const uint8_t key[],
+                                size_t key_len,
+                                void **pResult);
+        int AES_encrypt(const void *state,
+                        const uint8_t *in,
+                        uint8_t *out,
+                        size_t data_len);
+        int AES_decrypt(const void *state,
+                        const uint8_t *in,
+                        uint8_t *out,
+                        size_t data_len);
+        int AES_stop_operation(void *state);
+        """
+
+
+# Load portable AES
+_raw_aes_lib = load_pycryptodome_raw_lib("Crypto.Cipher._raw_aes",
+                                         _cproto)
+
+# Try to load AES with AES NI instructions
+try:
+    _raw_aesni_lib = None
+    if _cpu_features.have_aes_ni():
+        _raw_aesni_lib = load_pycryptodome_raw_lib("Crypto.Cipher._raw_aesni",
+                                                   _cproto.replace("AES",
+                                                                   "AESNI"))
+# _raw_aesni may not have been compiled in
+except OSError:
+    pass
+
+
+def _create_base_cipher(dict_parameters):
+    """This method instantiates and returns a handle to a low-level
+    base cipher. It will absorb named parameters in the process."""
+
+    use_aesni = dict_parameters.pop("use_aesni", True)
+
+    try:
+        key = dict_parameters.pop("key")
+    except KeyError:
+        raise TypeError("Missing 'key' parameter")
+
+    if len(key) not in key_size:
+        raise ValueError("Incorrect AES key length (%d bytes)" % len(key))
+
+    if use_aesni and _raw_aesni_lib:
+        start_operation = _raw_aesni_lib.AESNI_start_operation
+        stop_operation = _raw_aesni_lib.AESNI_stop_operation
+    else:
+        start_operation = _raw_aes_lib.AES_start_operation
+        stop_operation = _raw_aes_lib.AES_stop_operation
+
+    cipher = VoidPointer()
+    result = start_operation(c_uint8_ptr(key),
+                             c_size_t(len(key)),
+                             cipher.address_of())
+    if result:
+        raise ValueError("Error %X while instantiating the AES cipher"
+                         % result)
+    return SmartPointer(cipher.get(), stop_operation)
+
+
+def _derive_Poly1305_key_pair(key, nonce):
+    """Derive a tuple (r, s, nonce) for a Poly1305 MAC.
+
+    If nonce is ``None``, a new 16-byte nonce is generated.
+    """
+
+    if len(key) != 32:
+        raise ValueError("Poly1305 with AES requires a 32-byte key")
+
+    if nonce is None:
+        nonce = get_random_bytes(16)
+    elif len(nonce) != 16:
+        raise ValueError("Poly1305 with AES requires a 16-byte nonce")
+
+    s = new(key[:16], MODE_ECB).encrypt(nonce)
+    return key[16:], s, nonce
+
+
+def new(key, mode, *args, **kwargs):
+    """Create a new AES cipher.
+
+    Args:
+      key(bytes/bytearray/memoryview):
+        The secret key to use in the symmetric cipher.
+
+        It must be 16 (*AES-128)*, 24 (*AES-192*) or 32 (*AES-256*) bytes long.
+
+        For ``MODE_SIV`` only, it doubles to 32, 48, or 64 bytes.
+      mode (a ``MODE_*`` constant):
+        The chaining mode to use for encryption or decryption.
+        If in doubt, use ``MODE_EAX``.
+
+    Keyword Args:
+      iv (bytes/bytearray/memoryview):
+        (Only applicable for ``MODE_CBC``, ``MODE_CFB``, ``MODE_OFB``,
+        and ``MODE_OPENPGP`` modes).
+
+        The initialization vector to use for encryption or decryption.
+
+        For ``MODE_CBC``, ``MODE_CFB``, and ``MODE_OFB`` it must be 16 bytes long.
+
+        For ``MODE_OPENPGP`` mode only,
+        it must be 16 bytes long for encryption
+        and 18 bytes for decryption (in the latter case, it is
+        actually the *encrypted* IV which was prefixed to the ciphertext).
+
+        If not provided, a random byte string is generated (you must then
+        read its value with the :attr:`iv` attribute).
+
+      nonce (bytes/bytearray/memoryview):
+        (Only applicable for ``MODE_CCM``, ``MODE_EAX``, ``MODE_GCM``,
+        ``MODE_SIV``, ``MODE_OCB``, and ``MODE_CTR``).
+
+        A value that must never be reused for any other encryption done
+        with this key (except possibly for ``MODE_SIV``, see below).
+
+        For ``MODE_EAX``, ``MODE_GCM`` and ``MODE_SIV`` there are no
+        restrictions on its length (recommended: **16** bytes).
+
+        For ``MODE_CCM``, its length must be in the range **[7..13]**.
+        Bear in mind that with CCM there is a trade-off between nonce
+        length and maximum message size. Recommendation: **11** bytes.
+
+        For ``MODE_OCB``, its length must be in the range **[1..15]**
+        (recommended: **15**).
+
+        For ``MODE_CTR``, its length must be in the range **[0..15]**
+        (recommended: **8**).
+
+        For ``MODE_SIV``, the nonce is optional, if it is not specified,
+        then no nonce is being used, which renders the encryption
+        deterministic.
+
+        If not provided, for modes other than ``MODE_SIV``, a random
+        byte string of the recommended length is used (you must then
+        read its value with the :attr:`nonce` attribute).
+
+      segment_size (integer):
+        (Only ``MODE_CFB``).The number of **bits** the plaintext and ciphertext
+        are segmented in. It must be a multiple of 8.
+        If not specified, it will be assumed to be 8.
+
+      mac_len (integer):
+        (Only ``MODE_EAX``, ``MODE_GCM``, ``MODE_OCB``, ``MODE_CCM``)
+        Length of the authentication tag, in bytes.
+
+        It must be even and in the range **[4..16]**.
+        The recommended value (and the default, if not specified) is **16**.
+
+      msg_len (integer):
+        (Only ``MODE_CCM``). Length of the message to (de)cipher.
+        If not specified, ``encrypt`` must be called with the entire message.
+        Similarly, ``decrypt`` can only be called once.
+
+      assoc_len (integer):
+        (Only ``MODE_CCM``). Length of the associated data.
+        If not specified, all associated data is buffered internally,
+        which may represent a problem for very large messages.
+
+      initial_value (integer or bytes/bytearray/memoryview):
+        (Only ``MODE_CTR``).
+        The initial value for the counter. If not present, the cipher will
+        start counting from 0. The value is incremented by one for each block.
+        The counter number is encoded in big endian mode.
+
+      counter (object):
+        (Only ``MODE_CTR``).
+        Instance of ``Crypto.Util.Counter``, which allows full customization
+        of the counter block. This parameter is incompatible to both ``nonce``
+        and ``initial_value``.
+
+      use_aesni: (boolean):
+        Use Intel AES-NI hardware extensions (default: use if available).
+
+    Returns:
+        an AES object, of the applicable mode.
+    """
+
+    kwargs["add_aes_modes"] = True
+    return _create_cipher(sys.modules[__name__], key, mode, *args, **kwargs)
+
+
+# Size of a data block (in bytes)
+block_size = 16
+# Size of a key (in bytes)
+key_size = (16, 24, 32)

venv/lib/python3.11/site-packages/Crypto/Cipher/AES.pyi

@@ -0,0 +1,154 @@
+from typing import ByteString, Dict, Optional, Tuple, Union, overload
+from typing_extensions import Literal
+
+from Crypto.Cipher._mode_ecb import EcbMode
+from Crypto.Cipher._mode_cbc import CbcMode
+from Crypto.Cipher._mode_cfb import CfbMode
+from Crypto.Cipher._mode_ofb import OfbMode
+from Crypto.Cipher._mode_ctr import CtrMode
+from Crypto.Cipher._mode_openpgp import OpenPgpMode
+from Crypto.Cipher._mode_ccm import CcmMode
+from Crypto.Cipher._mode_eax import EaxMode
+from Crypto.Cipher._mode_gcm import GcmMode
+from Crypto.Cipher._mode_siv import SivMode
+from Crypto.Cipher._mode_ocb import OcbMode
+
+MODE_ECB: Literal[1]
+MODE_CBC: Literal[2]
+MODE_CFB: Literal[3]
+MODE_OFB: Literal[5]
+MODE_CTR: Literal[6]
+MODE_OPENPGP: Literal[7]
+MODE_CCM: Literal[8]
+MODE_EAX: Literal[9]
+MODE_SIV: Literal[10]
+MODE_GCM: Literal[11]
+MODE_OCB: Literal[12]
+
+# MODE_ECB
+@overload
+def new(key: ByteString,
+        mode: Literal[1],
+        use_aesni : bool = ...) -> \
+        EcbMode: ...
+
+# MODE_CBC
+@overload
+def new(key: ByteString,
+        mode: Literal[2],
+        iv : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        CbcMode: ...
+
+@overload
+def new(key: ByteString,
+        mode: Literal[2],
+        IV : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        CbcMode: ...
+
+# MODE_CFB
+@overload
+def new(key: ByteString,
+        mode: Literal[3],
+        iv : Optional[ByteString] = ...,
+        segment_size : int = ...,
+        use_aesni : bool = ...) -> \
+        CfbMode: ...
+
+@overload
+def new(key: ByteString,
+        mode: Literal[3],
+        IV : Optional[ByteString] = ...,
+        segment_size : int = ...,
+        use_aesni : bool = ...) -> \
+        CfbMode: ...
+
+# MODE_OFB
+@overload
+def new(key: ByteString,
+        mode: Literal[5],
+        iv : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        OfbMode: ...
+
+@overload
+def new(key: ByteString,
+        mode: Literal[5],
+        IV : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        OfbMode: ...
+
+# MODE_CTR
+@overload
+def new(key: ByteString,
+        mode: Literal[6],
+        nonce : Optional[ByteString] = ...,
+        initial_value : Union[int, ByteString] = ...,
+        counter : Dict = ...,
+        use_aesni : bool = ...) -> \
+        CtrMode: ...
+
+# MODE_OPENPGP
+@overload
+def new(key: ByteString,
+        mode: Literal[7],
+        iv : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        OpenPgpMode: ...
+
+@overload
+def new(key: ByteString,
+        mode: Literal[7],
+        IV : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        OpenPgpMode: ...
+
+# MODE_CCM
+@overload
+def new(key: ByteString,
+        mode: Literal[8],
+        nonce : Optional[ByteString] = ...,
+        mac_len : int = ...,
+        assoc_len : int = ...,
+        use_aesni : bool = ...) -> \
+        CcmMode: ...
+
+# MODE_EAX
+@overload
+def new(key: ByteString,
+        mode: Literal[9],
+        nonce : Optional[ByteString] = ...,
+        mac_len : int = ...,
+        use_aesni : bool = ...) -> \
+        EaxMode: ...
+
+# MODE_GCM
+@overload
+def new(key: ByteString,
+        mode: Literal[10],
+        nonce : Optional[ByteString] = ...,
+        use_aesni : bool = ...) -> \
+        SivMode: ...
+
+# MODE_SIV
+@overload
+def new(key: ByteString,
+        mode: Literal[11],
+        nonce : Optional[ByteString] = ...,
+        mac_len : int = ...,
+        use_aesni : bool = ...) -> \
+        GcmMode: ...
+
+# MODE_OCB
+@overload
+def new(key: ByteString,
+        mode: Literal[12],
+        nonce : Optional[ByteString] = ...,
+        mac_len : int = ...,
+        use_aesni : bool = ...) -> \
+        OcbMode: ...
+
+
+block_size: int
+key_size: Tuple[int, int, int]

venv/lib/python3.11/site-packages/Crypto/Cipher/ARC2.py

@@ -0,0 +1,175 @@
+# -*- coding: utf-8 -*-
+#
+#  Cipher/ARC2.py : ARC2.py
+#
+# ===================================================================
+# The contents of this file are dedicated to the public domain.  To
+# the extent that dedication to the public domain is not available,
+# everyone is granted a worldwide, perpetual, royalty-free,
+# non-exclusive license to exercise all rights associated with the
+# contents of this file for any purpose whatsoever.
+# No rights are reserved.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+# ===================================================================
+"""
+Module's constants for the modes of operation supported with ARC2:
+
+:var MODE_ECB: :ref:`Electronic Code Book (ECB) <ecb_mode>`
+:var MODE_CBC: :ref:`Cipher-Block Chaining (CBC) <cbc_mode>`
+:var MODE_CFB: :ref:`Cipher FeedBack (CFB) <cfb_mode>`
+:var MODE_OFB: :ref:`Output FeedBack (OFB) <ofb_mode>`
+:var MODE_CTR: :ref:`CounTer Mode (CTR) <ctr_mode>`
+:var MODE_OPENPGP:  :ref:`OpenPGP Mode <openpgp_mode>`
+:var MODE_EAX: :ref:`EAX Mode <eax_mode>`
+"""
+
+import sys
+
+from Crypto.Cipher import _create_cipher
+from Crypto.Util.py3compat import byte_string
+from Crypto.Util._raw_api import (load_pycryptodome_raw_lib,
+                                  VoidPointer, SmartPointer,
+                                  c_size_t, c_uint8_ptr)
+
+_raw_arc2_lib = load_pycryptodome_raw_lib(
+                        "Crypto.Cipher._raw_arc2",
+                        """
+                        int ARC2_start_operation(const uint8_t key[],
+                                                 size_t key_len,
+                                                 size_t effective_key_len,
+                                                 void **pResult);
+                        int ARC2_encrypt(const void *state,
+                                         const uint8_t *in,
+                                         uint8_t *out,
+                                         size_t data_len);
+                        int ARC2_decrypt(const void *state,
+                                         const uint8_t *in,
+                                         uint8_t *out,
+                                         size_t data_len);
+                        int ARC2_stop_operation(void *state);
+                        """
+                        )
+
+
+def _create_base_cipher(dict_parameters):
+    """This method instantiates and returns a handle to a low-level
+    base cipher. It will absorb named parameters in the process."""
+
+    try:
+        key = dict_parameters.pop("key")
+    except KeyError:
+        raise TypeError("Missing 'key' parameter")
+
+    effective_keylen = dict_parameters.pop("effective_keylen", 1024)
+
+    if len(key) not in key_size:
+        raise ValueError("Incorrect ARC2 key length (%d bytes)" % len(key))
+
+    if not (40 <= effective_keylen <= 1024):
+        raise ValueError("'effective_key_len' must be at least 40 and no larger than 1024 "
+                         "(not %d)" % effective_keylen)
+
+    start_operation = _raw_arc2_lib.ARC2_start_operation
+    stop_operation = _raw_arc2_lib.ARC2_stop_operation
+
+    cipher = VoidPointer()
+    result = start_operation(c_uint8_ptr(key),
+                             c_size_t(len(key)),
+                             c_size_t(effective_keylen),
+                             cipher.address_of())
+    if result:
+        raise ValueError("Error %X while instantiating the ARC2 cipher"
+                         % result)
+
+    return SmartPointer(cipher.get(), stop_operation)
+
+
+def new(key, mode, *args, **kwargs):
+    """Create a new RC2 cipher.
+
+    :param key:
+        The secret key to use in the symmetric cipher.
+        Its length can vary from 5 to 128 bytes; the actual search space
+        (and the cipher strength) can be reduced with the ``effective_keylen`` parameter.
+    :type key: bytes, bytearray, memoryview
+
+    :param mode:
+        The chaining mode to use for encryption or decryption.
+    :type mode: One of the supported ``MODE_*`` constants
+
+    :Keyword Arguments:
+        *   **iv** (*bytes*, *bytearray*, *memoryview*) --
+            (Only applicable for ``MODE_CBC``, ``MODE_CFB``, ``MODE_OFB``,
+            and ``MODE_OPENPGP`` modes).
+
+            The initialization vector to use for encryption or decryption.
+
+            For ``MODE_CBC``, ``MODE_CFB``, and ``MODE_OFB`` it must be 8 bytes long.
+
+            For ``MODE_OPENPGP`` mode only,
+            it must be 8 bytes long for encryption
+            and 10 bytes for decryption (in the latter case, it is
+            actually the *encrypted* IV which was prefixed to the ciphertext).
+
+            If not provided, a random byte string is generated (you must then
+            read its value with the :attr:`iv` attribute).
+
+        *   **nonce** (*bytes*, *bytearray*, *memoryview*) --
+            (Only applicable for ``MODE_EAX`` and ``MODE_CTR``).
+
+            A value that must never be reused for any other encryption done
+            with this key.
+
+            For ``MODE_EAX`` there are no
+            restrictions on its length (recommended: **16** bytes).
+
+            For ``MODE_CTR``, its length must be in the range **[0..7]**.
+
+            If not provided for ``MODE_EAX``, a random byte string is generated (you
+            can read it back via the ``nonce`` attribute).
+
+        *   **effective_keylen** (*integer*) --
+            Optional. Maximum strength in bits of the actual key used by the ARC2 algorithm.
+            If the supplied ``key`` parameter is longer (in bits) of the value specified
+            here, it will be weakened to match it.
+            If not specified, no limitation is applied.
+
+        *   **segment_size** (*integer*) --
+            (Only ``MODE_CFB``).The number of **bits** the plaintext and ciphertext
+            are segmented in. It must be a multiple of 8.
+            If not specified, it will be assumed to be 8.
+
+        *   **mac_len** : (*integer*) --
+            (Only ``MODE_EAX``)
+            Length of the authentication tag, in bytes.
+            It must be no longer than 8 (default).
+
+        *   **initial_value** : (*integer*) --
+            (Only ``MODE_CTR``). The initial value for the counter within
+            the counter block. By default it is **0**.
+
+    :Return: an ARC2 object, of the applicable mode.
+    """
+
+    return _create_cipher(sys.modules[__name__], key, mode, *args, **kwargs)
+
+MODE_ECB = 1
+MODE_CBC = 2
+MODE_CFB = 3
+MODE_OFB = 5
+MODE_CTR = 6
+MODE_OPENPGP = 7
+MODE_EAX = 9
+
+# Size of a data block (in bytes)
+block_size = 8
+# Size of a key (in bytes)
+key_size = range(5, 128 + 1)

venv/lib/python3.11/site-packages/Crypto/Cipher/ARC2.pyi

@@ -0,0 +1,33 @@
+from typing import Union, Dict, Iterable, Optional, ByteString
+
+from Crypto.Cipher._mode_ecb import EcbMode
+from Crypto.Cipher._mode_cbc import CbcMode
+from Crypto.Cipher._mode_cfb import CfbMode
+from Crypto.Cipher._mode_ofb import OfbMode
+from Crypto.Cipher._mode_ctr import CtrMode
+from Crypto.Cipher._mode_openpgp import OpenPgpMode
+from Crypto.Cipher._mode_eax import EaxMode
+
+ARC2Mode = int
+
+MODE_ECB: ARC2Mode
+MODE_CBC: ARC2Mode
+MODE_CFB: ARC2Mode
+MODE_OFB: ARC2Mode
+MODE_CTR: ARC2Mode
+MODE_OPENPGP: ARC2Mode
+MODE_EAX: ARC2Mode
+
+def new(key: ByteString,
+        mode: ARC2Mode,
+        iv : Optional[ByteString] = ...,
+        IV : Optional[ByteString] = ...,
+        nonce : Optional[ByteString] = ...,
+        segment_size : int = ...,
+        mac_len : int = ...,
+        initial_value : Union[int, ByteString] = ...,
+        counter : Dict = ...) -> \
+        Union[EcbMode, CbcMode, CfbMode, OfbMode, CtrMode, OpenPgpMode]: ...
+
+block_size: int
+key_size: Iterable[int]

venv/lib/python3.11/site-packages/Crypto/Cipher/ARC4.py

@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+#
+#  Cipher/ARC4.py : ARC4
+#
+# ===================================================================
+# The contents of this file are dedicated to the public domain.  To
+# the extent that dedication to the public domain is not available,
+# everyone is granted a worldwide, perpetual, royalty-free,
+# non-exclusive license to exercise all rights associated with the
+# contents of this file for any purpose whatsoever.
+# No rights are reserved.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+# ===================================================================
+
+from Crypto.Util._raw_api import (load_pycryptodome_raw_lib, VoidPointer,
+                                  create_string_buffer, get_raw_buffer,
+                                  SmartPointer, c_size_t, c_uint8_ptr)
+
+
+_raw_arc4_lib = load_pycryptodome_raw_lib("Crypto.Cipher._ARC4", """
+                    int ARC4_stream_encrypt(void *rc4State, const uint8_t in[],
+                                            uint8_t out[], size_t len);
+                    int ARC4_stream_init(uint8_t *key, size_t keylen,
+                                         void **pRc4State);
+                    int ARC4_stream_destroy(void *rc4State);
+                    """)
+
+
+class ARC4Cipher:
+    """ARC4 cipher object. Do not create it directly. Use
+    :func:`Crypto.Cipher.ARC4.new` instead.
+    """
+
+    def __init__(self, key, *args, **kwargs):
+        """Initialize an ARC4 cipher object
+
+        See also `new()` at the module level."""
+
+        if len(args) > 0:
+            ndrop = args[0]
+            args = args[1:]
+        else:
+            ndrop = kwargs.pop('drop', 0)
+
+        if len(key) not in key_size:
+            raise ValueError("Incorrect ARC4 key length (%d bytes)" %
+                             len(key))
+
+        self._state = VoidPointer()
+        result = _raw_arc4_lib.ARC4_stream_init(c_uint8_ptr(key),
+                                                c_size_t(len(key)),
+                                                self._state.address_of())
+        if result != 0:
+            raise ValueError("Error %d while creating the ARC4 cipher"
+                             % result)
+        self._state = SmartPointer(self._state.get(),
+                                   _raw_arc4_lib.ARC4_stream_destroy)
+
+        if ndrop > 0:
+            # This is OK even if the cipher is used for decryption,
+            # since encrypt and decrypt are actually the same thing
+            # with ARC4.
+            self.encrypt(b'\x00' * ndrop)
+
+        self.block_size = 1
+        self.key_size = len(key)
+
+    def encrypt(self, plaintext):
+        """Encrypt a piece of data.
+
+        :param plaintext: The data to encrypt, of any size.
+        :type plaintext: bytes, bytearray, memoryview
+        :returns: the encrypted byte string, of equal length as the
+          plaintext.
+        """
+
+        ciphertext = create_string_buffer(len(plaintext))
+        result = _raw_arc4_lib.ARC4_stream_encrypt(self._state.get(),
+                                                   c_uint8_ptr(plaintext),
+                                                   ciphertext,
+                                                   c_size_t(len(plaintext)))
+        if result:
+            raise ValueError("Error %d while encrypting with RC4" % result)
+        return get_raw_buffer(ciphertext)
+
+    def decrypt(self, ciphertext):
+        """Decrypt a piece of data.
+
+        :param ciphertext: The data to decrypt, of any size.
+        :type ciphertext: bytes, bytearray, memoryview
+        :returns: the decrypted byte string, of equal length as the
+          ciphertext.
+        """
+
+        try:
+            return self.encrypt(ciphertext)
+        except ValueError as e:
+            raise ValueError(str(e).replace("enc", "dec"))
+
+
+def new(key, *args, **kwargs):
+    """Create a new ARC4 cipher.
+
+    :param key:
+        The secret key to use in the symmetric cipher.
+        Its length must be in the range ``[1..256]``.
+        The recommended length is 16 bytes.
+    :type key: bytes, bytearray, memoryview
+
+    :Keyword Arguments:
+        *   *drop* (``integer``) --
+            The amount of bytes to discard from the initial part of the keystream.
+            In fact, such part has been found to be distinguishable from random
+            data (while it shouldn't) and also correlated to key.
+
+            The recommended value is 3072_ bytes. The default value is 0.
+
+    :Return: an `ARC4Cipher` object
+
+    .. _3072: http://eprint.iacr.org/2002/067.pdf
+    """
+    return ARC4Cipher(key, *args, **kwargs)
+
+
+# Size of a data block (in bytes)
+block_size = 1
+# Size of a key (in bytes)
+key_size = range(1, 256+1)

venv/lib/python3.11/site-packages/Crypto/Cipher/ARC4.pyi

@@ -0,0 +1,14 @@
+from typing import Any, Union, Iterable, ByteString
+
+class ARC4Cipher:
+    block_size: int
+    key_size: int
+
+    def __init__(self, key: ByteString, *args: Any, **kwargs: Any) -> None: ...
+    def encrypt(self, plaintext: ByteString) -> bytes: ...
+    def decrypt(self, ciphertext: ByteString) -> bytes: ...
+
+def new(key: ByteString, drop : int = ...) -> ARC4Cipher: ...
+
+block_size: int
+key_size: Iterable[int]

venv/lib/python3.11/site-packages/Crypto/Cipher/Blowfish.py

@@ -0,0 +1,159 @@
+# -*- coding: utf-8 -*-
+#
+#  Cipher/Blowfish.py : Blowfish
+#
+# ===================================================================
+# The contents of this file are dedicated to the public domain.  To
+# the extent that dedication to the public domain is not available,
+# everyone is granted a worldwide, perpetual, royalty-free,
+# non-exclusive license to exercise all rights associated with the
+# contents of this file for any purpose whatsoever.
+# No rights are reserved.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+# ===================================================================
+"""
+Module's constants for the modes of operation supported with Blowfish:
+
+:var MODE_ECB: :ref:`Electronic Code Book (ECB) <ecb_mode>`
+:var MODE_CBC: :ref:`Cipher-Block Chaining (CBC) <cbc_mode>`
+:var MODE_CFB: :ref:`Cipher FeedBack (CFB) <cfb_mode>`
+:var MODE_OFB: :ref:`Output FeedBack (OFB) <ofb_mode>`
+:var MODE_CTR: :ref:`CounTer Mode (CTR) <ctr_mode>`
+:var MODE_OPENPGP:  :ref:`OpenPGP Mode <openpgp_mode>`
+:var MODE_EAX: :ref:`EAX Mode <eax_mode>`
+"""
+
+import sys
+
+from Crypto.Cipher import _create_cipher
+from Crypto.Util._raw_api import (load_pycryptodome_raw_lib,
+                                  VoidPointer, SmartPointer, c_size_t,
+                                  c_uint8_ptr)
+
+_raw_blowfish_lib = load_pycryptodome_raw_lib(
+        "Crypto.Cipher._raw_blowfish",
+        """
+        int Blowfish_start_operation(const uint8_t key[],
+                                     size_t key_len,
+                                     void **pResult);
+        int Blowfish_encrypt(const void *state,
+                             const uint8_t *in,
+                             uint8_t *out,
+                             size_t data_len);
+        int Blowfish_decrypt(const void *state,
+                             const uint8_t *in,
+                             uint8_t *out,
+                             size_t data_len);
+        int Blowfish_stop_operation(void *state);
+        """
+        )
+
+
+def _create_base_cipher(dict_parameters):
+    """This method instantiates and returns a smart pointer to
+    a low-level base cipher. It will absorb named parameters in
+    the process."""
+
+    try:
+        key = dict_parameters.pop("key")
+    except KeyError:
+        raise TypeError("Missing 'key' parameter")
+
+    if len(key) not in key_size:
+        raise ValueError("Incorrect Blowfish key length (%d bytes)" % len(key))
+
+    start_operation = _raw_blowfish_lib.Blowfish_start_operation
+    stop_operation = _raw_blowfish_lib.Blowfish_stop_operation
+
+    void_p = VoidPointer()
+    result = start_operation(c_uint8_ptr(key),
+                             c_size_t(len(key)),
+                             void_p.address_of())
+    if result:
+        raise ValueError("Error %X while instantiating the Blowfish cipher"
+                         % result)
+    return SmartPointer(void_p.get(), stop_operation)
+
+
+def new(key, mode, *args, **kwargs):
+    """Create a new Blowfish cipher
+
+    :param key:
+        The secret key to use in the symmetric cipher.
+        Its length can vary from 5 to 56 bytes.
+    :type key: bytes, bytearray, memoryview
+
+    :param mode:
+        The chaining mode to use for encryption or decryption.
+    :type mode: One of the supported ``MODE_*`` constants
+
+    :Keyword Arguments:
+        *   **iv** (*bytes*, *bytearray*, *memoryview*) --
+            (Only applicable for ``MODE_CBC``, ``MODE_CFB``, ``MODE_OFB``,
+            and ``MODE_OPENPGP`` modes).
+
+            The initialization vector to use for encryption or decryption.
+
+            For ``MODE_CBC``, ``MODE_CFB``, and ``MODE_OFB`` it must be 8 bytes long.
+
+            For ``MODE_OPENPGP`` mode only,
+            it must be 8 bytes long for encryption
+            and 10 bytes for decryption (in the latter case, it is
+            actually the *encrypted* IV which was prefixed to the ciphertext).
+
+            If not provided, a random byte string is generated (you must then
+            read its value with the :attr:`iv` attribute).
+
+        *   **nonce** (*bytes*, *bytearray*, *memoryview*) --
+            (Only applicable for ``MODE_EAX`` and ``MODE_CTR``).
+
+            A value that must never be reused for any other encryption done
+            with this key.
+
+            For ``MODE_EAX`` there are no
+            restrictions on its length (recommended: **16** bytes).
+
+            For ``MODE_CTR``, its length must be in the range **[0..7]**.
+
+            If not provided for ``MODE_EAX``, a random byte string is generated (you
+            can read it back via the ``nonce`` attribute).
+
+        *   **segment_size** (*integer*) --
+            (Only ``MODE_CFB``).The number of **bits** the plaintext and ciphertext
+            are segmented in. It must be a multiple of 8.
+            If not specified, it will be assumed to be 8.
+
+        *   **mac_len** : (*integer*) --
+            (Only ``MODE_EAX``)
+            Length of the authentication tag, in bytes.
+            It must be no longer than 8 (default).
+
+        *   **initial_value** : (*integer*) --
+            (Only ``MODE_CTR``). The initial value for the counter within
+            the counter block. By default it is **0**.
+
+    :Return: a Blowfish object, of the applicable mode.
+    """
+
+    return _create_cipher(sys.modules[__name__], key, mode, *args, **kwargs)
+
+MODE_ECB = 1
+MODE_CBC = 2
+MODE_CFB = 3
+MODE_OFB = 5
+MODE_CTR = 6
+MODE_OPENPGP = 7
+MODE_EAX = 9
+
+# Size of a data block (in bytes)
+block_size = 8
+# Size of a key (in bytes)
+key_size = range(4, 56 + 1)

venv/lib/python3.11/site-packages/Crypto/Cipher/Blowfish.pyi

@@ -0,0 +1,33 @@
+from typing import Union, Dict, Iterable, ByteString, Optional
+
+from Crypto.Cipher._mode_ecb import EcbMode
+from Crypto.Cipher._mode_cbc import CbcMode
+from Crypto.Cipher._mode_cfb import CfbMode
+from Crypto.Cipher._mode_ofb import OfbMode
+from Crypto.Cipher._mode_ctr import CtrMode
+from Crypto.Cipher._mode_openpgp import OpenPgpMode
+from Crypto.Cipher._mode_eax import EaxMode
+
+BlowfishMode = int
+
+MODE_ECB: BlowfishMode
+MODE_CBC: BlowfishMode
+MODE_CFB: BlowfishMode
+MODE_OFB: BlowfishMode
+MODE_CTR: BlowfishMode
+MODE_OPENPGP: BlowfishMode
+MODE_EAX: BlowfishMode
+
+def new(key: ByteString,
+        mode: BlowfishMode,
+        iv : Optional[ByteString] = ...,
+        IV : Optional[ByteString] = ...,
+        nonce : Optional[ByteString] = ...,
+        segment_size : int = ...,
+        mac_len : int = ...,
+        initial_value : Union[int, ByteString] = ...,
+        counter : Dict = ...) -> \
+        Union[EcbMode, CbcMode, CfbMode, OfbMode, CtrMode, OpenPgpMode]: ...
+
+block_size: int
+key_size: Iterable[int]

venv/lib/python3.11/site-packages/Crypto/Cipher/CAST.py

@@ -0,0 +1,159 @@
+# -*- coding: utf-8 -*-
+#
+#  Cipher/CAST.py : CAST
+#
+# ===================================================================
+# The contents of this file are dedicated to the public domain.  To
+# the extent that dedication to the public domain is not available,
+# everyone is granted a worldwide, perpetual, royalty-free,
+# non-exclusive license to exercise all rights associated with the
+# contents of this file for any purpose whatsoever.
+# No rights are reserved.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+# ===================================================================
+"""
+Module's constants for the modes of operation supported with CAST:
+
+:var MODE_ECB: :ref:`Electronic Code Book (ECB) <ecb_mode>`
+:var MODE_CBC: :ref:`Cipher-Block Chaining (CBC) <cbc_mode>`
+:var MODE_CFB: :ref:`Cipher FeedBack (CFB) <cfb_mode>`
+:var MODE_OFB: :ref:`Output FeedBack (OFB) <ofb_mode>`
+:var MODE_CTR: :ref:`CounTer Mode (CTR) <ctr_mode>`
+:var MODE_OPENPGP:  :ref:`OpenPGP Mode <openpgp_mode>`
+:var MODE_EAX: :ref:`EAX Mode <eax_mode>`
+"""
+
+import sys
+
+from Crypto.Cipher import _create_cipher
+from Crypto.Util.py3compat import byte_string
+from Crypto.Util._raw_api import (load_pycryptodome_raw_lib,
+                                  VoidPointer, SmartPointer,
+                                  c_size_t, c_uint8_ptr)
+
+_raw_cast_lib = load_pycryptodome_raw_lib(
+                    "Crypto.Cipher._raw_cast",
+                    """
+                    int CAST_start_operation(const uint8_t key[],
+                                             size_t key_len,
+                                             void **pResult);
+                    int CAST_encrypt(const void *state,
+                                     const uint8_t *in,
+                                     uint8_t *out,
+                                     size_t data_len);
+                    int CAST_decrypt(const void *state,
+                                     const uint8_t *in,
+                                     uint8_t *out,
+                                     size_t data_len);
+                    int CAST_stop_operation(void *state);
+                    """)
+
+
+def _create_base_cipher(dict_parameters):
+    """This method instantiates and returns a handle to a low-level
+    base cipher. It will absorb named parameters in the process."""
+
+    try:
+        key = dict_parameters.pop("key")
+    except KeyError:
+        raise TypeError("Missing 'key' parameter")
+
+    if len(key) not in key_size:
+        raise ValueError("Incorrect CAST key length (%d bytes)" % len(key))
+
+    start_operation = _raw_cast_lib.CAST_start_operation
+    stop_operation = _raw_cast_lib.CAST_stop_operation
+
+    cipher = VoidPointer()
+    result = start_operation(c_uint8_ptr(key),
+                             c_size_t(len(key)),
+                             cipher.address_of())
+    if result:
+        raise ValueError("Error %X while instantiating the CAST cipher"
+                         % result)
+
+    return SmartPointer(cipher.get(), stop_operation)
+
+
+def new(key, mode, *args, **kwargs):
+    """Create a new CAST cipher
+
+    :param key:
+        The secret key to use in the symmetric cipher.
+        Its length can vary from 5 to 16 bytes.
+    :type key: bytes, bytearray, memoryview
+
+    :param mode:
+        The chaining mode to use for encryption or decryption.
+    :type mode: One of the supported ``MODE_*`` constants
+
+    :Keyword Arguments:
+        *   **iv** (*bytes*, *bytearray*, *memoryview*) --
+            (Only applicable for ``MODE_CBC``, ``MODE_CFB``, ``MODE_OFB``,
+            and ``MODE_OPENPGP`` modes).
+
+            The initialization vector to use for encryption or decryption.
+
+            For ``MODE_CBC``, ``MODE_CFB``, and ``MODE_OFB`` it must be 8 bytes long.
+
+            For ``MODE_OPENPGP`` mode only,
+            it must be 8 bytes long for encryption
+            and 10 bytes for decryption (in the latter case, it is
+            actually the *encrypted* IV which was prefixed to the ciphertext).
+
+            If not provided, a random byte string is generated (you must then
+            read its value with the :attr:`iv` attribute).
+
+        *   **nonce** (*bytes*, *bytearray*, *memoryview*) --
+            (Only applicable for ``MODE_EAX`` and ``MODE_CTR``).
+
+            A value that must never be reused for any other encryption done
+            with this key.
+
+            For ``MODE_EAX`` there are no
+            restrictions on its length (recommended: **16** bytes).
+
+            For ``MODE_CTR``, its length must be in the range **[0..7]**.
+
+            If not provided for ``MODE_EAX``, a random byte string is generated (you
+            can read it back via the ``nonce`` attribute).
+
+        *   **segment_size** (*integer*) --
+            (Only ``MODE_CFB``).The number of **bits** the plaintext and ciphertext
+            are segmented in. It must be a multiple of 8.
+            If not specified, it will be assumed to be 8.
+
+        *   **mac_len** : (*integer*) --
+            (Only ``MODE_EAX``)
+            Length of the authentication tag, in bytes.
+            It must be no longer than 8 (default).
+
+        *   **initial_value** : (*integer*) --
+            (Only ``MODE_CTR``). The initial value for the counter within
+            the counter block. By default it is **0**.
+
+    :Return: a CAST object, of the applicable mode.
+    """
+
+    return _create_cipher(sys.modules[__name__], key, mode, *args, **kwargs)
+
+MODE_ECB = 1
+MODE_CBC = 2
+MODE_CFB = 3
+MODE_OFB = 5
+MODE_CTR = 6
+MODE_OPENPGP = 7
+MODE_EAX = 9
+
+# Size of a data block (in bytes)
+block_size = 8
+# Size of a key (in bytes)
+key_size = range(5, 16 + 1)

venv/lib/python3.11/site-packages/Crypto/Cipher/CAST.pyi

@@ -0,0 +1,33 @@
+from typing import Union, Dict, Iterable, Optional, ByteString
+
+from Crypto.Cipher._mode_ecb import EcbMode
+from Crypto.Cipher._mode_cbc import CbcMode
+from Crypto.Cipher._mode_cfb import CfbMode
+from Crypto.Cipher._mode_ofb import OfbMode
+from Crypto.Cipher._mode_ctr import CtrMode
+from Crypto.Cipher._mode_openpgp import OpenPgpMode
+from Crypto.Cipher._mode_eax import EaxMode
+
+CASTMode = int
+
+MODE_ECB: CASTMode
+MODE_CBC: CASTMode
+MODE_CFB: CASTMode
+MODE_OFB: CASTMode
+MODE_CTR: CASTMode
+MODE_OPENPGP: CASTMode
+MODE_EAX: CASTMode
+
+def new(key: ByteString,
+        mode: CASTMode,
+        iv : Optional[ByteString] = ...,
+        IV : Optional[ByteString] = ...,
+        nonce : Optional[ByteString] = ...,
+        segment_size : int = ...,
+        mac_len : int = ...,
+        initial_value : Union[int, ByteString] = ...,
+        counter : Dict = ...) -> \
+        Union[EcbMode, CbcMode, CfbMode, OfbMode, CtrMode, OpenPgpMode]: ...
+
+block_size: int
+key_size : Iterable[int]