Update app.py

app.py

@@ -11,13 +11,39 @@ def chatgpt(query):
         output = "Please try: https://chat.openai.com"
     return output
 
+def generateimage1(query):
+    try:
+        output = dictionary[query]
+    except:
+        output = "Please try: https://chat.openai.com"
+    return output
+
+def generateimage2(query):
+    try:
+        output = dictionary[query]
+    except:
+        output = "Please try: https://chat.openai.com"
+    return output
+
 with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
+    gr.Markdown("Ask ChatGPT to generate detailed descriptbions.")
+    with gr.Row():
+        inpprompt = gr.Dropdown(["a playful man", "a trusing woman", "a boy who is bullied"])
+        outprompt = gr.Textbox()
+    btn = gr.Button("Generate mode detailed description.")
+    btn.click(fn=chatgpt, inputs=inpprompt, outputs=outprompt)
+
+    
+    gr.Markdown("Ask **Stable Diffusion** to generate images.")
+    with gr.Row():
+        out1image = gr.Image()
+        out2image = gr.Image()
     with gr.Row():
-        inp = gr.Dropdown(["a playful man", "a trusing woman", "a boy who is bullied"])
-        out = gr.Textbox()
-    btn = gr.Button("Ask ChatGPT to generate a detailed description.")
-    btn.click(fn=chatgpt, inputs=inp, outputs=out)
+        btn1 = gr.Button("Generate image with origial prompt.")
+        btn2 = gr.Button("Generate image with ChatGPT's prompt.")
+        
+    btn.click(fn=generateimage1, inputs=inpprompt, outputs=out1image)
+    btn.click(fn=generateimage2, inputs=outprompt, outputs=out2image)
     
 
 demo.launch()