Upload app.py with huggingface_hub

app.py

@@ -5,9 +5,8 @@ from PIL import Image
 from io import BytesIO
 import replicate
 from dotenv import load_dotenv
-
-# Para la nueva versión de OpenAI
-from openai import OpenAI
+import json
+import time
 
 # Page configuration must be the first Streamlit command
 st.set_page_config(page_title="CuentAI – AI Story Maker", layout="wide")
@@ -19,6 +18,80 @@ load_dotenv()
 replicate_token = os.getenv("REPLICATE_API_TOKEN")
 openai_api_key = os.getenv("OPENAI_API_KEY")
 
+# Funciones para comunicarse con OpenAI directamente a través de requests en lugar de usar el SDK
+def openai_chat_completion(prompt, model="gpt-4-turbo-preview", temperature=0.8, max_tokens=600):
+    """Usa la API de OpenAI directamente a través de requests para evitar problemas con el SDK"""
+    
+    # Mostrar información para debugging
+    st.info(f"Usando clave de API: {openai_api_key[:5]}...{openai_api_key[-4:]}")
+    
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai_api_key}"
+    }
+    
+    payload = {
+        "model": model,
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": float(temperature),
+        "max_tokens": int(max_tokens)
+    }
+    
+    try:
+        # Imprimir información para depuración
+        st.write("Enviando solicitud a OpenAI...")
+        
+        # Imprimir payload para depuración (solo durante desarrollo)
+        st.json(payload)
+        
+        response = requests.post(
+            "https://api.openai.com/v1/chat/completions",
+            headers=headers,
+            json=payload,
+            timeout=60  # Aumentar timeout para evitar errores por tiempo
+        )
+        
+        st.write(f"Estado de respuesta: {response.status_code}")
+        
+        # Si hay un error, mostrar el detalle
+        if response.status_code != 200:
+            st.error(f"Error de API: {response.status_code} - {response.text}")
+            return None
+            
+        response_data = response.json()
+        st.success("Respuesta recibida correctamente")
+        return response_data["choices"][0]["message"]["content"]
+    except Exception as e:
+        st.error(f"Error con la API de OpenAI: {str(e)}")
+        return None
+
+def openai_generate_image(prompt):
+    """Genera imágenes con DALL-E 3 directamente a través de requests"""
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai_api_key}"
+    }
+    
+    payload = {
+        "model": "dall-e-3",
+        "prompt": prompt,
+        "n": 1,
+        "size": "1024x1024",
+        "quality": "standard"
+    }
+    
+    try:
+        response = requests.post(
+            "https://api.openai.com/v1/images/generations",
+            headers=headers,
+            json=payload
+        )
+        response.raise_for_status()
+        return response.json()["data"][0]["url"]
+    except Exception as e:
+        st.error(f"Error generando imagen: {e}")
+        return None
+
 # Optional TTS setup
 # Uncomment if using Google Cloud TTS
 # from google.cloud import texttospeech
@@ -56,16 +129,20 @@ def generate_story(name: str, theme: str) -> str:
     # Format the prompt with user inputs
     prompt = prompt_template.format(name=name, theme=theme)
     
+    # Mostrar mensaje de carga
+    with st.spinner("Generando tu historia personalizada..."): 
+        # Información de depuración
+        st.write("Datos de entrada:")
+        st.write(f"- Protagonista: {name}")
+        st.write(f"- Tema: {theme}")
+    
     try:
-        # Usar la nueva versión de la API de OpenAI (v1.x)
-        client = OpenAI(api_key=openai_api_key)
-        response = client.chat.completions.create(
-            model="gpt-4",
-            messages=[{"role": "user", "content": prompt}],
-            temperature=0.8,
-            max_tokens=600,
-        )
-        return response.choices[0].message.content.strip()
+        # Usar nuestra función personalizada que se comunica directamente con la API
+        story = openai_chat_completion(prompt)
+        if story:
+            return story.strip()
+        else:
+            raise Exception("No se pudo generar la historia")
     except Exception as e:
         st.error(f"Error generating story: {e}")
         # Return a fallback story for demo purposes
@@ -136,14 +213,12 @@ def generate_image_dalle(prompt: str, protagonist: str) -> str:
     )
     
     try:
-        # Usar la nueva versión de la API de OpenAI (v1.x)
-        client = OpenAI(api_key=openai_api_key)
-        response = client.images.generate(
-            prompt=full_prompt,
-            n=1,
-            size="512x512"
-        )
-        return response.data[0].url
+        # Usar nuestra función personalizada que se comunica directamente con la API
+        image_url = openai_generate_image(full_prompt)
+        if image_url:
+            return image_url
+        else:
+            raise Exception("No se pudo generar la imagen")
     except Exception as e:
         st.error(f"Error generating image: {e}")
         # Return a placeholder image URL