Update app.py

app.py

@@ -1,35 +1,44 @@
-from PIL import Image, ImageDraw, ImageFont
-
-# Dimensões da imagem
-width, height = 576, 1024
-
-# Criar uma imagem branca
-image = Image.new('RGB', (width, height), color='white')
-
-# Criar um objeto de desenho
-draw = ImageDraw.Draw(image)
-
-# Carregar uma fonte (certifique-se de que a fonte está disponível no seu sistema)
-try:
-    font = ImageFont.truetype("arial.ttf", 40)
-except IOError:
-    font = ImageFont.load_default()
-
-# Texto a ser adicionado à imagem
-text = "Este é um exemplo de texto em uma imagem."
-
-# Calcular a posição do texto para centralizá-lo
-bbox = draw.textbbox((0, 0), text, font=font)
-text_width = bbox[2] - bbox[0]
-text_height = bbox[3] - bbox[1]
-text_x = (width - text_width) / 2
-text_y = (height - text_height) / 2
-
-# Adicionar texto à imagem
-draw.text((text_x, text_y), text, font=font, fill='black')
-
-# Salvar a imagem
-image.save('text_image.png')
-
-# Exibir a imagem
-image.show()
+import requests
+import json
+
+# Defina a URL da API do Hugging Face
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
+
+# Defina o cabeçalho da requisição com o token de autenticação
+headers = {
+    "Authorization": "Bearer YOUR_HUGGINGFACE_API_TOKEN"
+}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response
+
+def generate_image(text_prompt, output_file="generated_image.png"):
+    data = {
+        "inputs": text_prompt,
+        "options": {
+            "wait_for_model": True,
+            "use_cpu": True,
+            "cpu_basic": {
+                "vCPU": 2,
+                "RAM": 16
+            }
+        }
+    }
+
+    response = query(data)
+
+    if response.status_code == 200:
+        with open(output_file, "wb") as f:
+            f.write(response.content)
+        print(f"Imagem gerada com sucesso e salva como '{output_file}'")
+    else:
+        print(f"Erro na requisição: {response.status_code}")
+        print(f"Mensagem de erro: {response.text}")
+
+if __name__ == "__main__":
+    # Defina o texto que será usado para gerar a imagem
+    text_prompt = "A beautiful sunset over the mountains"
+
+    # Gere a imagem
+    generate_image(text_prompt)