Update app.py

app.py

@@ -1,12 +1,42 @@
 import os
 import requests
 from geopy.geocoders import GoogleV3
-from text2image import generate_image
 import streamlit as st
 
 API_KEY = os.getenv('WEATHER_API')
 GOOGLE_API = os.getenv('GOOGLE_API')
 geolocator = GoogleV3(api_key=GOOGLE_API)
+STABLE_API = os.getenv("STABLE_API")
+
+def generate_image(prompt):
+  url = "https://stablediffusionapi.com/api/v3/text2img"
+
+  payload = json.dumps({
+    "key": STABLE_API,
+    "prompt": prompt,
+    "negative_prompt": None,
+    "width": "512",
+    "height": "512",
+    "samples": "1",
+    "num_inference_steps": "20",
+    "seed": None,
+    "guidance_scale": 7.5,
+    "safety_checker": "yes",
+    "multi_lingual": "no",
+    "panorama": "no",
+    "self_attention": "no",
+    "upscale": "no",
+    "embeddings_model": None,
+    "webhook": None,
+    "track_id": None
+  })
+
+  headers = {
+    'Content-Type': 'application/json'
+  }
+
+  response = requests.request("POST", url, headers=headers, data=payload)
+  return response.json()["output"][0]
 
 st.title("Weather Image")
 
@@ -19,7 +49,6 @@ submit = st.button("Submit", type="primary")
 # CITY = 'London'
 # url = f"https://api.openweathermap.org/data/2.5/weather?q={CITY}&appid={API_KEY}"
 
-
 if user_input and submit:
     location = geolocator.geocode(user_input)
     url = f"https://api.openweathermap.org/data/2.5/weather?lat={location.latitude}&lon={location.longitude}&appid={API_KEY}"