a

.devcontainer/devcontainer.json

@@ -0,0 +1,33 @@
+{
+  "name": "Python 3",
+  // Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
+  "image": "mcr.microsoft.com/devcontainers/python:1-3.11-bullseye",
+  "customizations": {
+    "codespaces": {
+      "openFiles": [
+        "README.md",
+        "images.py"
+      ]
+    },
+    "vscode": {
+      "settings": {},
+      "extensions": [
+        "ms-python.python",
+        "ms-python.vscode-pylance"
+      ]
+    }
+  },
+  "updateContentCommand": "[ -f packages.txt ] && sudo apt update && sudo apt upgrade -y && sudo xargs apt install -y <packages.txt; [ -f requirements.txt ] && pip3 install --user -r requirements.txt; pip3 install --user streamlit; echo '✅ Packages installed and Requirements met'",
+  "postAttachCommand": {
+    "server": "streamlit run images.py --server.enableCORS false --server.enableXsrfProtection false"
+  },
+  "portsAttributes": {
+    "8501": {
+      "label": "Application",
+      "onAutoForward": "openPreview"
+    }
+  },
+  "forwardPorts": [
+    8501
+  ]
+}

.gitignore

@@ -0,0 +1,16 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Secrets
+*.env
+
+# Other stuffs
+.devcotainer/
+images/
+# Virtual environments
+.venv

.python-version

@@ -0,0 +1 @@
+3.11

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import subprocess
+import random
 from urllib.parse import quote
 from io import BytesIO
 
@@ -38,17 +39,21 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-st.markdown("<h1 class='title-text'>🎨 AI Image Generator</h1>", unsafe_allow_html=True)
+st.markdown("<h1 style='background: linear-gradient(90deg, #ff7eb3, #ff758c); -webkit-background-clip: text; -webkit-text-fill-color: transparent;' class='title-text'>🎨 AI Image Generator</h1>", unsafe_allow_html=True)
 
 with st.sidebar:
     st.header("⚙️ Settings")
 
-    width = st.slider("Width:", 256, 2048, 1024, step=64)
-    height = st.slider("Height:", 256, 2048, 1024, step=64)
+    width = st.slider("Width:", 256, 2048, 512, step=1)
+    height = st.slider("Height:", 256, 2048, 512, step=1)
 
-    if "seed" not in st.session_state:
-        st.session_state.seed = 42  # Default seed
-    seed = st.number_input("Seed:", value=st.session_state.seed, min_value=0, step=1)
+    random_seed = st.checkbox("Use Random Seed", value=True)
+    if random_seed:
+        seed = random.randint(1, 1337)
+    else:
+        if "seed" not in st.session_state:
+            st.session_state.seed = 42  
+        seed = st.number_input("Seed:", value=st.session_state.seed, min_value=1, max_value=1337, step=1)
 
     model = st.selectbox("Model:", ["flux", "flux-pro", "flux-realism", "flux-anime", "flux-3d", "flux-cablyai", "turbo"], index=0)
 
@@ -56,12 +61,16 @@ with st.sidebar:
     nologo = st.checkbox("Remove Watermark", value=True)
     private = st.checkbox("Keep image private", value=True)
     enhance = st.checkbox("Enhance prompt", value=True)
+    safe = st.checkbox("Safe", value=True)
 
     st.subheader("ℹ️ About")
     st.markdown("""
-    [![Powered by Pollinations.ai](https://img.shields.io/badge/Powered%20by-Pollinations.ai-blue?style=for-the-badge&logo=appveyor&borderRadius=50)](https://pollinations.ai)  
+    [![Powered by Pollinations.ai](https://img.shields.io/badge/Powered%20by-Pollinations.ai-blue?style=for-the-badge&logo=rocket&borderRadius=50)](https://pollinations.ai)  
     [![Source Code](https://img.shields.io/badge/Source%20Code-GitHub-black?style=for-the-badge&logo=github&borderRadius=50)](https://github.com/oopshnik/image_gen)  
-    [![Developer](https://img.shields.io/badge/Developer-Contact-green?style=for-the-badge&logo=telegram&borderRadius=50)](https://t.me/pr_ogr)
+    [![Hugging Face Spaces](https://img.shields.io/badge/Hugging%20Face-Spaces-blue?style=for-the-badge&logo=huggingface&borderRadius=50)](https://huggingface.co/spaces/oopshnik/image_gen)
+    \n
+    [![Developer](https://img.shields.io/badge/Developer-Contact-green?style=for-the-badge&logo=telegram&borderRadius=50)](https://t.me/pr_ogr) 
+      
     """, unsafe_allow_html=True)
 
 st.markdown("<h3 style='text-align: center;'>Enter Your Image Prompt</h3>", unsafe_allow_html=True)
@@ -69,7 +78,6 @@ col1, col2, col3 = st.columns([1, 3, 1])
 with col2:
     prompt = st.text_input("Prompt:", placeholder="A futuristic city at sunset 🌇", label_visibility="collapsed")
 
-
 col1, col2, col3 = st.columns([1, 2, 1])
 
 with col2:
@@ -87,24 +95,16 @@ with col2:
                 "private": str(private).lower(),
                 "enhance": str(enhance).lower(),
                 "model": model,
-                "safe": "true"
+                "safe": str(safe).lower()
             }
             url = f"{base_url}{encoded_prompt}?" + "&".join([f"{k}={v}" for k, v in params.items()])
-
+        
             response = requests.get(url)
 
             if response.status_code == 200:
-                img_bytes = BytesIO(response.content)
-                process = subprocess.run(["curl", "-f", "-X", "POST", "-F", "file=@-", "https://0x0.st"],
-                        input=img_bytes.getvalue(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-
-                if process.returncode == 0:
-                    upload_url = process.stdout.decode().strip()
-
-
-                st.image(response.content, caption=f"✨ Generated Image \n🔗 {upload_url}", use_container_width=True)
-                st.success(f"✅ Image generated successfully! Seed used: {seed} | Width: {width} | Height: {height}\n {upload_url}")
-
 
+                st.image(response.content, caption=f"✨ Generated Image", width=width)
+                st.download_button('📥 Download Image', data=response.content, file_name=prompt+".png", mime="image/png")
+                st.code("")
             else:
-                st.error(f"❌ Failed to generate image. Status code: {response.status_code}")
+                st.error(f"❌ Failed to generate image. Status code: {response.status_code}")