initial commit

app.py

@@ -0,0 +1,104 @@
+import requests
+import io
+from PIL import Image
+import gradio as gr
+from together import Together
+from groq import Groq  # Import the Groq library
+
+# Initialize the Together and Groq clients with your API keys
+client = Together(api_key="d43d9d8b78242356408329a19a9cdce9290a8c3a254db6293fa6293d207e7e38")
+groq_client = Groq(api_key="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")
+
+def enhance_prompt(user_prompt):
+    """Enhances the given prompt using Groq and returns the refined prompt."""
+    try:
+        chat_completion = groq_client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": (
+                        "Create prompts that paint a clear picture for image generation. "
+                        "Be precise, detailed, and direct, describing not only the content of the image "
+                        "but also details like tone, style, color palette, and point of view. "
+                        "For photorealistic images, include the device used (e.g., 'shot on iPhone 16'), "
+                        "aperture, lens, and shot type. Use clear, visual descriptions. "
+                        "Prompt Structure: 'A [medium] of [subject], [subject’s characteristics], "
+                        "[relation to background] [background]. [Details of background] "
+                        "[Interactions with color and lighting]. (\"Taken on:\"/\"Drawn with:\")[Specific traits of style]' "
+                        "Include details on medium, subject (clothing, hairstyle, pose, etc.), background, "
+                        "colors, lighting, style traits, influences, technique, and camera settings if applicable."
+                    ),
+                },
+                {"role": "user", "content": user_prompt}
+            ],
+            model="llama-3.3-70b-versatile",
+            temperature=0.5,
+            max_completion_tokens=1024,
+            top_p=1,
+            stop=None,
+            stream=False,
+        )
+        enhanced = chat_completion.choices[0].message.content
+    except Exception as e:
+        enhanced = f"Error enhancing prompt: {str(e)}"
+    return enhanced
+
+def generate_image(prompt):
+    """Generates an image using the refined prompt."""
+    try:
+        response = client.images.generate(
+            prompt=prompt,
+            model="black-forest-labs/FLUX.1-dev-lora",
+            width=1024,
+            height=768,
+            steps=28,
+            n=1,
+            response_format="url",
+            image_loras=[
+                {"path": "https://huggingface.co/XLabs-AI/flux-RealismLora", "scale": 1},
+            ],
+        )
+        image_url = response.data[0].url
+        image_response = requests.get(image_url)
+        image_bytes = image_response.content
+        img = Image.open(io.BytesIO(image_bytes))
+    except Exception as e:
+        # Optionally, handle errors (you can also return a default error image)
+        img = None
+    return img
+
+# Build the Gradio interface with a two-step process
+with gr.Blocks(css=".gradio-container {background-color: #f9f9f9; padding: 20px;}") as demo:
+    gr.Markdown("# 2-Step Image Generator")
+    gr.Markdown(
+        "### Step 1: Prompt Enhancement\n"
+        "Enter your original prompt below and click **Enhance Prompt**. "
+        "The system will generate a detailed version of your prompt. You can modify the enhanced prompt before generating the image."
+    )
+    
+    with gr.Row():
+        original_prompt = gr.Textbox(
+            label="Your Original Prompt",
+            placeholder="Describe your idea here...",
+            lines=3
+        )
+        enhance_button = gr.Button("Enhance Prompt")
+    
+    enhanced_prompt_box = gr.Textbox(
+        label="Enhanced Prompt (Editable)",
+        placeholder="The enhanced prompt will appear here...",
+        lines=3
+    )
+    
+    enhance_button.click(fn=enhance_prompt, inputs=original_prompt, outputs=enhanced_prompt_box)
+    
+    gr.Markdown("### Step 2: Image Generation\n"
+                "Review and modify the enhanced prompt if necessary. Once you're ready, click **Generate Image** to create your image.")
+    
+    generate_button = gr.Button("Generate Image")
+    image_output = gr.Image(label="Generated Image")
+    
+    generate_button.click(fn=generate_image, inputs=enhanced_prompt_box, outputs=image_output)
+
+if __name__ == "__main__":
+    demo.launch()