Update app.py

app.py

@@ -14,7 +14,8 @@ import PIL
 import PIL.Image
 import PIL.ImageDraw
 import PIL.ImageFont
-from ChatPodcastGPT import Chat
+import gradio as gr
+
 
 logger = structlog.getLogger()
 weather_api_key = os.environ['WEATHER_API']
@@ -179,23 +180,37 @@ class WeatherDraw:
         logger.info(f"Got animal {animal}")
         chat = Chat(f'''
 Given the following weather conditions, write a plaintext, short, and vivid description of an
-adorable {animal} in the weather conditions doing an activity matching the weather.
+adorable {animal} in the weather conditions doing a human activity matching the weather.
 Only write the short description and nothing else.
 Do not include specific numbers.'''.replace('\n', ' '))
         description = chat.message(str(weather_info))
         prompt = f'{description}. Adorable, cute, 4k, Award winning, in the style of {random.choice(art_styles)}'
         logger.info(prompt)
         img = Image.create(prompt, **kwargs)
-        return img["b64_json"]
+        return img["b64_json"], prompt
 
     def step_one_forecast(self, weather_info, **kwargs):
-        img = self.generate_image(weather_info, **kwargs)
+        img, txt = self.generate_image(weather_info, **kwargs)
         # text = self.clean_text(weather_info)
         # return overlay_text_on_image(img, text, 'bottom-left')
-        return img
+        return img, txt
 
     def step(self, zip_code='10001', **kwargs):
         forecast = Weather(zip_code).get_info()
-        return {time: overlay_text_on_image(self.step_one_forecast(data, **kwargs), time, 'top-right') for time, data in forecast.items()}
-
-
+        images, texts = [], []
+        for time, data in forecast.items():
+            img, txt = overlay_text_on_image(self.step_one_forecast(data, **kwargs), time, 'top-right')
+            images.append(img)
+            texts.append(txt)
+        return create_collage(*images), *texts
+
+
+# Define Gradio interface
+iface = gr.Interface(fn=WeatherDraw().step, 
+                     inputs=gr.inputs.Text(label="Enter Zipcode"), 
+                     outputs=[gr.outputs.Image(), "text", "text", "text", "text"],
+                     title="US Zipcode Weather",
+                     description="Enter a US Zipcode and get some weather.")
+
+# Run the interface
+iface.launch()