Upload app.py with huggingface_hub

app.py

@@ -1,5 +1,5 @@
 import os
-import openai
+from openai import OpenAI
 import streamlit as st
 import requests
 from PIL import Image
@@ -12,7 +12,7 @@ st.set_page_config(page_title="CuentAI – AI Story Maker", layout="wide")
 
 # Load environment variables
 load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# El cliente OpenAI se inicializa en cada función para evitar problemas de compatibilidad
 replicate_token = os.getenv("REPLICATE_API_TOKEN")
 
 # Optional TTS setup
@@ -20,13 +20,16 @@ replicate_token = os.getenv("REPLICATE_API_TOKEN")
 # from google.cloud import texttospeech
 
 # Function to load prompt templates
-def load_prompt(file_path):
+def load_prompt(file_path, default_prompt=""):
+    """Cargar un archivo de prompt o devolver un valor predeterminado si falla"""
     try:
         with open(file_path, 'r', encoding='utf-8') as file:
             return file.read().strip()
     except Exception as e:
-        st.error(f"Error loading prompt file: {e}")
-        return ""
+        # No mostrar error si tenemos un valor predeterminado
+        if not default_prompt:
+            st.warning(f"No se pudo cargar el archivo {file_path}. Usando prompt predeterminado.")
+        return default_prompt
 
 # Story generation function
 def generate_story(name: str, theme: str) -> str:
@@ -48,7 +51,9 @@ def generate_story(name: str, theme: str) -> str:
     prompt = prompt_template.format(name=name, theme=theme)
     
     try:
-        response = openai.ChatCompletion.create(
+        # Usar la nueva versión de la API de OpenAI (v1.x)
+        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        response = client.chat.completions.create(
             model="gpt-4",
             messages=[{"role": "user", "content": prompt}],
             temperature=0.8,
@@ -120,12 +125,14 @@ def generate_image_dalle(prompt: str, protagonist: str) -> str:
     )
     
     try:
-        response = openai.Image.create(
+        # Usar la nueva versión de la API de OpenAI (v1.x)
+        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+        response = client.images.generate(
             prompt=full_prompt,
             n=1,
             size="512x512"
         )
-        return response["data"][0]["url"]
+        return response.data[0].url
     except Exception as e:
         st.error(f"Error generating image: {e}")
         # Return a placeholder image URL