Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import requests | |
| import base64 | |
| import os | |
| import time | |
| # Get your Hugging Face access token from the secrets | |
| HUGGING_FACE_API_KEY = os.getenv("onteddu") # Ensure your access token is properly set | |
| def generate_images_from_text(text): | |
| # Generate an image using Hugging Face's API | |
| headers = { | |
| "Authorization": f"Bearer {HUGGING_FACE_API_KEY}" | |
| } | |
| response = requests.post( | |
| "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell", # Correct model name | |
| headers=headers, | |
| json={"inputs": text} | |
| ) | |
| if response.status_code == 200: | |
| # The API response contains the image as raw binary data | |
| image_data = response.content | |
| # Convert the image to Base64 for embedding in HTML | |
| base64_image = base64.b64encode(image_data).decode('utf-8') | |
| return f"data:image/png;base64,{base64_image}" | |
| else: | |
| print(f"Error generating image: {response.text}") | |
| return None | |
| def generate_comic(short_story): | |
| sentences = short_story.split('. ') | |
| images = [] | |
| for sentence in sentences: | |
| if sentence.strip(): | |
| try: | |
| # Generate one image for each sentence | |
| image_data = generate_images_from_text(sentence) | |
| if image_data: | |
| images.append(image_data) | |
| # Sleep for a short duration to avoid hitting rate limits | |
| time.sleep(2) # Adjust this time if needed | |
| except Exception as e: | |
| print(f"Error generating image: {e}") | |
| # Create comic HTML with reduced image sizes and panel layout | |
| comic_html = '<div style="display: flex; flex-wrap: wrap;">' | |
| for img in images: | |
| comic_html += f'<div style="flex: 1; margin: 5px;"><img src="{img}" style="max-width: 300px; max-height: 300px;" /></div>' | |
| comic_html += '</div>' | |
| return comic_html | |
| # Gradio interface | |
| iface = gr.Interface( | |
| fn=generate_comic, | |
| inputs=gr.Textbox(label="Your Short Story", lines=10, placeholder="Enter your short story here..."), | |
| outputs=gr.HTML(label="Generated Comic"), | |
| title="Narrative to Comic Generator", | |
| description="Enter a short story, and the app will generate a comic-style storybook with images and dialogues." | |
| ) | |
| iface.launch() |