Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,4 @@
 import os
-from openai import OpenAI
 import streamlit as st
 import requests
 from PIL import Image
@@ -7,13 +6,18 @@ from io import BytesIO
 import replicate
 from dotenv import load_dotenv
 
+# Para la nueva versión de OpenAI
+from openai import OpenAI
+
 # Page configuration must be the first Streamlit command
 st.set_page_config(page_title="CuentAI – AI Story Maker", layout="wide")
 
 # Load environment variables
 load_dotenv()
-# El cliente OpenAI se inicializa en cada función para evitar problemas de compatibilidad
+
+# Configuración de APIs
 replicate_token = os.getenv("REPLICATE_API_TOKEN")
+openai_api_key = os.getenv("OPENAI_API_KEY")
 
 # Optional TTS setup
 # Uncomment if using Google Cloud TTS
@@ -38,21 +42,23 @@ def generate_story(name: str, theme: str) -> str:
     with protagonist {name} and theme {theme}. Use playful tone,
     simple dialogue, and a clear beginning, middle, and end.
     """
-    # Load prompt template
-    prompt_template = load_prompt("prompts/story_prompt.txt")
-    if not prompt_template:
-        prompt_template = (
-            f"You are a children's story author. "
-            f"Write a 300-400 word children's story where the protagonist is named {name} "
-            f"and the plot is about {theme}. Use a friendly style and simple dialogue."
-        )
+    # Prompt predeterminado integrado en caso de que fallen los archivos externos
+    default_template = (
+        "You are a children's story author. "
+        "Write a 300-400 word children's story where the protagonist is named {name} "
+        "and the plot is about {theme}. Use a friendly style and simple dialogue. "
+        "Include a clear beginning, middle, and end."
+    )
+    
+    # Intentar cargar desde archivo, con fallback al predeterminado
+    prompt_template = load_prompt("prompts/story_prompt.txt", default_template)
     
     # Format the prompt with user inputs
     prompt = prompt_template.format(name=name, theme=theme)
     
     try:
         # Usar la nueva versión de la API de OpenAI (v1.x)
-        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        client = OpenAI(api_key=openai_api_key)
         response = client.chat.completions.create(
             model="gpt-4",
             messages=[{"role": "user", "content": prompt}],
@@ -113,10 +119,15 @@ def generate_image_dalle(prompt: str, protagonist: str) -> str:
     Call OpenAI Image API to create one 512×512 image from the prompt.
     Returns the image URL.
     """
-    # Load image prompt template
-    img_prompt_template = load_prompt("prompts/image_prompt.txt")
-    if not img_prompt_template:
-        img_prompt_template = "Crea una ilustración de estilo infantil y colorido para un cuento para niños. La escena muestra: {scene_description} Con {protagonist_name} como personaje principal."
+    # Prompt predeterminado para generación de imágenes
+    default_template = (
+        "Create a colorful children's book style illustration showing: {scene_description} " 
+        "With {protagonist_name} as the main character. Use bright colors, simple shapes, " 
+        "and a child-friendly art style."
+    )
+    
+    # Intentar cargar el prompt desde archivo con fallback al predeterminado
+    img_prompt_template = load_prompt("prompts/image_prompt.txt", default_template)
     
     # Format the prompt with user inputs
     full_prompt = img_prompt_template.format(
@@ -126,7 +137,7 @@ def generate_image_dalle(prompt: str, protagonist: str) -> str:
     
     try:
         # Usar la nueva versión de la API de OpenAI (v1.x)
-        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        client = OpenAI(api_key=openai_api_key)
         response = client.images.generate(
             prompt=full_prompt,
             n=1,