Update app.py

app.py

@@ -1,10 +1,12 @@
 import streamlit as st
-from freeGPT import Client
+from freeGPT import Client  # Asegúrate de que esta es la importación correcta
 from PIL import Image
 from io import BytesIO
 
-# Configurar el cliente de freeGPT
-client = Client()
+# Si el cliente no se puede instanciar, usa la función adecuada para crear la generación
+def generate_image(prompt):
+    # Aquí debes llamar a la función adecuada de freeGPT para generar la imagen
+    return Client.create_generation("MODEL", prompt)  # Ajusta esto si es necesario
 
 # Título de la aplicación
 st.title("Generador de Imágenes con AI")
@@ -16,7 +18,7 @@ if st.button("Generar Imagen"):
     if prompt:
         try:
             # Generar la imagen usando el prompt
-            resp = client.create_generation("prodia", prompt)
+            resp = generate_image(prompt)
             image = Image.open(BytesIO(resp))
 
             # Mostrar la imagen en la aplicación
@@ -25,4 +27,4 @@ if st.button("Generar Imagen"):
         except Exception as e:
             st.error(f"🤖: Ocurrió un error: {e}")
     else:
-        st.warning("Por favor, ingresa un prompt: ")
+        st.warning("Por favor, ingresa un prompt.")