Initial commit

.gitignore

@@ -0,0 +1,3 @@
+config.yaml
+.streamlit/*
+__pycache__/*

app.py

@@ -0,0 +1,29 @@
+from model import generate_image, generate_image_prompt, process_image_prompt
+import streamlit as st
+from streamlit_utlis import check_password
+
+if not check_password():
+    st.stop()
+
+st.title('Image Generator🐶')
+description = st.text_input(r'Describe the image you want. I will give you a few prompt to choose 🐻‍❄️')
+prompt = ''
+
+if st.button('GENERATE PROMPT 🐯'):
+    st.success('Generating prompts for you, please wait... 🦁')
+    response = generate_image_prompt(description)
+    result = process_image_prompt(response)
+
+    st.subheader("Here are some prompts for you to use🐧")
+    for i in range(len(result)):
+        st.text(result[i].replace("-", str(i+1)))
+
+st.divider()
+
+prompt = st.text_input(r"Get a prompt from above or enter your own 🦝")
+if st.button('GENERATE IMAGE 🐥'):
+    st.success('Generating image for you, please wait... 🐨')
+    result = generate_image(prompt)
+
+    st.subheader("Here is your image... 🦭")
+    st.image(result, caption=prompt)

model.py

@@ -0,0 +1,67 @@
+from openai import AzureOpenAI
+import json
+from io import BytesIO
+import requests
+import re
+import streamlit as st
+
+def get_client():
+    client = AzureOpenAI(
+        api_version="2024-05-01-preview",
+        azure_endpoint=st.secrets['endpoint'],
+        api_key=st.secrets['key'],
+    )
+    return client
+
+def generate_image(prompt):
+    client = get_client()
+    result = client.images.generate(
+        model="Dalle3",
+        prompt=prompt,
+        n=1
+    )
+
+    image_url = json.loads(result.model_dump_json())['data'][0]['url']
+    result = requests.get(image_url)
+    return BytesIO(result.content)
+
+def generate_image_prompt(prompt):
+    payload = {
+        "messages": [
+            {
+            "role": "system",
+            "content": [
+                {
+                "type": "text",
+                "text": "You give a few examples of english prompts that help generate image base on user's input. Return prompts in bullet point"
+                }
+            ]
+            },
+            {
+            "role": "user",
+            "content": [
+                {
+                "type": "text",
+                "text": prompt
+                }
+            ]
+            }
+        ],
+        "temperature": 0.9,
+        "top_p": 0.95,
+        "max_tokens": 800
+    }
+
+    response = requests.post(st.secrets['completionendpoint'], headers={"Content-Type": "application/json", "api-key": st.secrets['key']}, json=payload)
+    response.raise_for_status()  # Will raise an HTTPError if the HTTP request returned an unsuccessful status code
+    return response.json()['choices'][0]['message']['content']
+
+def process_image_prompt(response):
+    response = response.split('\n')
+    response = [re.sub(r"(?<!\\)['\"](.*?)(?<!\\)['\"]", r"\1", response[i]) for i in range(len(response))]
+    return response
+
+if __name__ == "__main__":
+    response = generate_image_prompt('halong bay, vietnam')
+    response = process_image_prompt(response)
+    

requirements.txt

@@ -0,0 +1,4 @@
+openai==1.43.1
+pillow==10.4.0
+requests==2.32.3
+streamlit==1.38.0

streamlit_utlis.py

@@ -0,0 +1,20 @@
+import hmac
+import streamlit as st
+
+def check_password():
+    def password_entered():
+        if hmac.compare_digest(st.session_state["password"], st.secrets["adminpassword"]):
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # Don't store the password.
+        else:
+            st.session_state["password_correct"] = False
+
+    if st.session_state.get("password_correct", False):
+        return True
+    
+    st.text_input(
+        "Enter Password 🚀", type="password", on_change=password_entered, key="password"
+    )
+    if "password_correct" in st.session_state:
+        st.error("Password incorrect 😕")
+    return False