Update app.py

app.py

@@ -8,12 +8,6 @@ from transformers import GPT2Tokenizer
 import requests
 import time
 
-
-# define azure openai context
-
-openai.api_type = "azure"
-openai.api_version = "2023-03-15-preview"
-
 def text_prompt(page_url, contraseña, temp, azure_endpoint):
     # Reemplazar estas líneas con cadenas de texto fijas
     request = """Analizar el siguiente texto de una noticia en prensa y generar un informe tipo KYC (Know Your Customer) para análisis de riesgos, considerando los siguientes aspectos:
@@ -66,9 +60,14 @@ def text_prompt(page_url, contraseña, temp, azure_endpoint):
     tokenization_time = time.time() - start_time
     print(f"Tokenization time: {tokenization_time:.4f} seconds")
 
-    if num_tokens > 10 and contraseña.startswith("sk-"):
+    if num_tokens > 10:
+        # define azure openai context
+        openai.api_type = "azure"
+        openai.api_base = azure_endpoint
+        openai.api_version = "2023-03-15-preview"
         openai.api_key = contraseña
-        # get the response from openai API
+        
+        # get the response from Azure OpenAI API
         start_time = time.time()
         try:
             response = openai.ChatCompletion.create(