Update python-chatGPT course

Signed-off-by: Manuel Vergara <manuel@vergaracarmona.es>

python-chatgpt/code/miPrograma.py

@@ -1,5 +1,6 @@
 import os
 import openai
+import spacy
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -13,22 +14,51 @@ openai.api_key = api_key
 # print(modelos)
 
 modelo = "text-davinci-002"
-prompt = "Elige un buen nombre para un elefante"
+prompt = "Cuenta una historia breve de algún lugar concreto"
 
 respuesta = openai.Completion.create(
     engine=modelo,
     prompt=prompt,
-    n=3,  # Opcional. Número de respuestas
+    n=1,  # Opcional. Número de respuestas
     temperature=1,  # Opcional. Controla la creatividad de la respuesta
-    max_tokens=50  # Opcional. Número máximo de tokens en la respuesta
+    max_tokens=200  # Opcional. Número máximo de tokens en la respuesta
 )
 
 # print(respuesta)
 
-# texto_generado = respuesta['choices'][0]['text']
-# print(texto_generado)
+texto_generado = respuesta.choices[0].text.strip()
+print(texto_generado)
 
 
-for idx, opcion in enumerate(respuesta.choices):
-    texto_generado = opcion.text.strip()
-    print(f"Respuesta {idx + 1}: {texto_generado}\n")
+# for idx, opcion in enumerate(respuesta.choices):
+#     texto_generado = opcion.text.strip()
+#     print(f"Respuesta {idx + 1}: {texto_generado}\n")
+
+print("***")
+
+modelo_spacy = spacy.load("es_core_news_md")
+
+analisis = modelo_spacy(texto_generado)
+
+# for token in analisis:
+#     print(token.text, token.pos_, token.dep_, token.head.text)
+
+ubicacion = None
+
+for ent in analisis.ents:
+    # print(ent.text, ent.label_)
+    if ent.label_ == "LOC":
+        ubicacion = ent
+        break
+
+if ubicacion:
+    prompt2 = f"Dime más acerca de {ubicacion}"
+    respuesta2 = openai.Completion.create(
+        engine=modelo,
+        prompt=prompt2,
+        n=1,
+        temperature=1,
+        max_tokens=100
+    )
+
+    print(respuesta2.choices[0].text.strip())