Upload 10 files

.gitignore

@@ -0,0 +1,179 @@
+# Created by https://www.toptal.com/developers/gitignore/api/python
+# Edit at https://www.toptal.com/developers/gitignore?templates=python
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+### Python Patch ###
+# Poetry local configuration file - https://python-poetry.org/docs/configuration/#local-configuration
+poetry.toml
+
+# ruff
+.ruff_cache/
+
+# LSP config files
+pyrightconfig.json
+
+# End of https://www.toptal.com/developers/gitignore/api/python
+
+.env
+.streamlit/secrets.toml

.streamlit/secrets.toml.example

@@ -0,0 +1,2 @@
+CLOUDFLARE_ACCOUNT_ID = "Your account id"
+CLOUDFLARE_API_TOKEN = "Your API key"

Inicio.py

@@ -0,0 +1,21 @@
+import streamlit as st
+
+st.set_page_config(
+    page_title="Cloudflare AI Playground",
+    page_icon="pages/media/icon.png"
+)
+
+"""
+# Generacion de imagenes con IA 
+
+---
+Generar de imagenes con AI mediante prompt
+
+Edita de imagenes con IA
+
+Genera imagenes apartir de otra imagen
+
+----
+"""
+
+st.markdown("[sixthz](https://instagram.com/sixthxz) was here",unsafe_allow_html=True)

README.md

@@ -1,13 +1,25 @@
----
-title: Tti
-emoji: 💻
-colorFrom: blue
-colorTo: blue
-sdk: streamlit
-sdk_version: 1.39.0
-app_file: app.py
-pinned: false
-short_description: CloudFlare Workers
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Generador texto a imagen usando los workers de cloudflare
+
+## Instalacion
+
+Copiar [.streamlit/secrets.toml.example](./.streamlit/secrets.toml.example) a `.streamlit/secrets.toml`.
+
+```bash
+python -m venv venv
+source ./venv/bin/activate
+python -m pip install -r requirements.txt
+```
+
+```cmd
+python -m venv venv
+venv\Scripts\activate
+python -m pip install -r requirements.txt
+```
+
+## Correr
+
+Correr app de Streamlit:
+
+```bash
+python -m streamlit run Inicio.py
+```

pages/1_Texto a Imagen.py

@@ -0,0 +1,39 @@
+import requests
+import streamlit as st
+
+account_id = st.secrets["CLOUDFLARE_ACCOUNT_ID"]
+api_token = st.secrets["CLOUDFLARE_API_TOKEN"]
+
+st.set_page_config(
+    page_title="Texto a Imagen",
+    page_icon="media/icon.png"
+)
+
+" ## Texto a Imagen"
+"""
+---
+"""
+
+with st.form("text_to_image"):
+    model = st.selectbox(
+        "Elige un modelo",
+        options=(
+            "@cf/lykon/dreamshaper-8-lcm",
+            "@cf/bytedance/stable-diffusion-xl-lightning",
+            "@cf/stabilityai/stable-diffusion-xl-base-1.0",
+        ),
+    )
+    prompt = st.text_area(label="Describe de manera detallada la imagen que deseas generar")
+    submitted = st.form_submit_button("Generar")
+    if submitted:
+        headers = {
+            "Authorization": f"Bearer {api_token}",
+        }
+        with st.spinner("Generando..."):
+            url = f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/{model}"
+            response = requests.post(
+                url,
+                headers=headers,
+                json={"prompt": prompt},
+            )
+            st.image(response.content, caption=prompt)        

pages/2_Editar Imagen.py

@@ -0,0 +1,67 @@
+from base64 import b64encode
+import io
+
+from PIL import Image, ImageOps
+import requests
+import streamlit as st
+from streamlit_drawable_canvas import st_canvas
+
+api_token = st.secrets["CLOUDFLARE_API_TOKEN"]
+account_id = st.secrets["CLOUDFLARE_ACCOUNT_ID"]
+
+st.set_page_config(
+    page_title="Editar Imagen",
+    page_icon="media/icon.png"
+)
+
+" ## Editar imagen"
+"""
+---
+"""
+
+image_upload = st.file_uploader("Sube una imagen")
+
+if not image_upload:
+    st.stop()
+
+img = Image.open(image_upload)
+img = ImageOps.contain(img, (600, 600))
+masking_result = st_canvas(width=img.width, height=img.height, background_image=img)
+
+def image_to_int_array(image, format="PNG"):
+    """Current Workers AI REST API consumes an array of unsigned 8 bit integers"""
+    bytes = io.BytesIO()
+    image.save(bytes, format=format)
+    return list(bytes.getvalue())
+
+if masking_result.image_data is not None:
+    # st.image(masking_result.image_data, caption="Your masking")
+    with st.form("Prompt"):
+        prompt = st.text_input(label="Que te gustaría reemplazar de esta imagen?")
+        submitted = st.form_submit_button("Generar")
+        if submitted:
+            model = "@cf/runwayml/stable-diffusion-v1-5-inpainting"
+            image_array = image_to_int_array(img)
+            # Reverse the order / Boolean array to mark the bits
+            mask = masking_result.image_data[:, :, -1] > 0
+            mask_image = Image.fromarray(mask)
+            mask_array = image_to_int_array(mask_image)
+            with st.spinner("Generando..."):
+                url = f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/{model}"
+                response = requests.post(
+                    url,
+                    headers={
+                        "Authorization": f"Bearer {api_token}",
+                    },
+                    json={"prompt": prompt, "image": image_array, "mask": mask_array},
+                )
+                if response.ok:
+                    st.image(response.content, caption=prompt)
+                else:
+                    st.warning(f"Error {response.status_code}")
+                    st.warning(response.reason)
+                    st.warning(response.text)
+                    f"Image Array is {len(image_array)} entries, first 10:"
+                    st.code(image_array[:10])
+                    f"Mask Array is {len(mask_array)} entries, first 10:"
+                    st.code(mask_array[:10])

pages/3_Imagen a Imagen.py

@@ -0,0 +1,66 @@
+from base64 import b64encode
+import io
+
+from PIL import Image, ImageOps
+import requests
+import streamlit as st
+import numpy as np
+
+api_token = st.secrets["CLOUDFLARE_API_TOKEN"]
+account_id = st.secrets["CLOUDFLARE_ACCOUNT_ID"]
+
+st.set_page_config(
+    page_title="Imagen a Imagen",
+    page_icon="media/icon.png"
+)
+
+" ## Imagen a imagen"
+st.markdown("*Ajustando parametros del modelo para mejores resultados")
+"""
+---
+"""
+
+# Cargar la imagen
+input_image = st.file_uploader("Sube una imagen")
+if not input_image:
+    st.stop()
+
+img = Image.open(input_image)
+img = ImageOps.contain(img, (600, 600))
+
+# Crear una máscara blanca que cubra toda la imagen
+masking_result = np.ones((img.height, img.width), dtype=np.uint8) * 255  # Máscara blanca
+
+def image_to_int_array(image, format="PNG"):
+    """Convertir la imagen a un array de enteros."""
+    bytes = io.BytesIO()
+    image.save(bytes, format=format)
+    return list(bytes.getvalue())
+
+with st.form("Prompt"):
+    prompt = st.text_input(label="Describe las modificaciones que quieres")
+    submitted = st.form_submit_button("Generar")
+    if submitted:
+        model = "@cf/runwayml/stable-diffusion-v1-5-img2img"
+        image_array = image_to_int_array(img)
+
+        # Crear la máscara a partir del masking_result
+        mask_array = image_to_int_array(Image.fromarray(masking_result))
+
+        with st.spinner("Generando..."):
+            url = f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/{model}"
+            response = requests.post(
+                url,
+                headers={
+                    "Authorization": f"Bearer {api_token}",
+                },
+                json={"prompt": prompt, "image": image_array, "mask": mask_array, "strength": 0.65, "guidance": 15},
+            )
+            if response.ok:
+                st.image(response.content, caption=prompt)
+            else:
+                st.warning(f"Error {response.status_code}")
+                st.warning(response.reason)
+                st.warning(response.text)
+                st.code(image_array[:10])
+                st.code(mask_array[:10])

requirements.txt

@@ -0,0 +1,4 @@
+requests
+streamlit
+streamlit-drawable-canvas==0.8.0
+watchdog