balmung commited on
Commit
08ab654
·
1 Parent(s): 6138b29

[ADD] retrieve picture from Gladia based of user input

Browse files
Files changed (1) hide show
  1. app.py +27 -5
app.py CHANGED
@@ -1,9 +1,31 @@
1
- from gen_num import *
2
  import streamlit as st
 
 
 
 
 
3
 
4
  st.write("""
5
- # TEST text
6
- hello there
7
  """)
8
- if (st.button("zzzzzzz")):
9
- print(generate_number("AI_generated_blurred.png"))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
+ import requests
3
+ from PIL import Image
4
+
5
+ URL = 'https://v2-api.gladia.io/text/image/image-generation/?model=stable-diffusion'
6
+ API_KEY = '37feed03-3d70-4ce7-aee6-d0c2e398b3a3'
7
 
8
  st.write("""
9
+ # LavAIamp
10
+ Do you have the hots for TRUE randomness ?
11
  """)
12
+
13
+ prompt = st.text_input('Keygen prompt', 'kirby', 420)
14
+ seed = st.number_input('Keygen value', 120, 1200000000)
15
+ steps = st.number_input('Steps number', 40, 100)
16
+
17
+ def gen_picture_gladia():
18
+ global prompt
19
+ global seed
20
+ try:
21
+ res = requests.post(URL, headers={'x-gladia-key': API_KEY}, data={"prompt": prompt, "integer": 1, "steps": steps, "seed": seed})
22
+ st.write(res.status_code)
23
+ with open("picc.png", 'wb') as f:
24
+ f.write(res.content)
25
+ image = Image.open('picc.png')
26
+ st.image(image, caption='Your newly created picture')
27
+ except:
28
+ st.write("ERROR while sending data to Gladia")
29
+
30
+ if st.button("Submit Data"):
31
+ gen_picture_gladia()