Delete app.py

app.py

@@ -1,110 +0,0 @@
-import gradio as gr
-from together import Together
-from PIL import Image
-import requests
-from io import BytesIO
-import os
-from datetime import datetime
-
-# Retrieve API key from Hugging Face Secrets
-API_KEY = os.getenv("MY_API_KEY")
-if not API_KEY:
-    print("Error: API_KEY not found. Please check your Hugging Face secrets.")
-client = Together(api_key=API_KEY)
-
-# Ensure directory for saving user-generated images
-IMAGE_FOLDER = "./user_images"
-os.makedirs(IMAGE_FOLDER, exist_ok=True)
-
-# Function to generate an image and save it
-def generate_and_save_image(prompt):
-    try:
-        if not prompt.strip():
-            print("Empty prompt received.")
-            return None, None  # Avoid empty prompts
-        
-        print(f"Generating image for: {prompt}")
-
-        # Generate image with Together AI API
-        response = client.images.generate(
-            prompt=prompt,
-            model="black-forest-labs/FLUX.1-schnell",  # Model name
-            steps=4
-        )
-        print(f"Response: {response}")
-
-        # Fetch the image from the returned URL
-        image_url = response.data[0].url
-        print(f"Image URL: {image_url}")
-        response_image = requests.get(image_url)
-        image = Image.open(BytesIO(response_image.content))
-
-        # Save the image locally
-        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
-        image_path = os.path.join(IMAGE_FOLDER, f"{timestamp}.png")
-        image.save(image_path)
-        print(f"Image successfully saved at: {image_path}")
-
-        return image, image_path  # Return the generated image and its path
-    except Exception as e:
-        print("Error during image generation or saving:", e)
-        return None, None
-
-# Function to get all saved images for the gallery
-def get_gallery_images():
-    try:
-        images = []
-        for filename in sorted(os.listdir(IMAGE_FOLDER)):
-            if filename.endswith(".png"):
-                images.append(os.path.join(IMAGE_FOLDER, filename))
-        print(f"Gallery images: {images}")
-        return images
-    except Exception as e:
-        print("Error fetching gallery images:", e)
-        return []
-
-# Function to zip all images and provide for download
-def download_gallery():
-    from shutil import make_archive
-    zip_path = os.path.join(IMAGE_FOLDER, "gallery")
-    make_archive(zip_path, 'zip', IMAGE_FOLDER)
-    return f"{zip_path}.zip"
-
-# Gradio Interface Setup
-with gr.Blocks() as app:
-    gr.Markdown("## Real-Time AI Image Generator 🚀")
-    gr.Markdown("Start typing a prompt below to generate images in real-time with Together AI's FLUX model. Your creations will be displayed below in the gallery.")
-
-    with gr.Row():
-        prompt_box = gr.Textbox(label="Enter your prompt", placeholder="A horse on the moon")
-        image_output = gr.Image(label="Generated Image")
-
-    # Add a gallery section
-    with gr.Row():
-        gallery = gr.Gallery(label="Generated with Love by You", height="auto")
-
-    # Add a download button for the gallery
-    with gr.Row():
-        download_button = gr.File(label="Download Your Gallery")
-    
-    def process_and_update_gallery(prompt):
-        image, image_path = generate_and_save_image(prompt)
-        gallery_images = get_gallery_images()
-        logging.debug(f"Updating gallery with {len(gallery_images)} images")
-        return image, gallery_images
-
-    # Trigger zip file creation for download
-    def prepare_download():
-        return download_gallery()
-    
-    prompt_box.change(
-        fn=process_and_update_gallery,
-        inputs=prompt_box,
-        outputs=[image_output, gallery]
-    )
-
-    # Bind download button to zip creation
-    download_button.change(fn=prepare_download, inputs=[], outputs=download_button)
-
-# Launch the Gradio app
-app.launch()