Update app.py

app.py

@@ -14,39 +14,30 @@ def chatgpt(query):
     return output
 
 def generateimage1(query):
-    try:
-        output = dictionary[query]
-    except:
-        output = "Please try: https://chat.openai.com"
-    return output
+    image = query.replace(' ', '-') + '.png'
+    return image
 
-def generateimage2(query):
-    try:
-        output = dictionary[query]
-    except:
-        output = "Please try: https://chat.openai.com"
-    return output
+def generateimage1(query):
+    image = query.replace(' ', '-') + '-chatgpt.png'
+    return image
 
 with gr.Blocks() as demo:
     gr.Markdown("Ask **ChatGPT** to generate detailed descriptbions.")
     gr.Markdown("Example: Act as a professional character designer. Please generate prompts include detail appearance descriptions that related to the personality. Here's your first question: 'a gentle man':")
+    gr.Markdown("Then ask **Stable Diffusion** to generate images based on ChatGPT prompts.")
     with gr.Row():
         inpprompt = gr.Dropdown(list(dictionary.keys()))
         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():
-        btn1 = gr.Button("Generate image with origial prompt.")
-        btn2 = gr.Button("Generate image with ChatGPT's prompt.")
-        
-    btn1.click(fn=generateimage1, inputs=inpprompt, outputs=out1image)
-    btn2.click(fn=generateimage2, inputs=outprompt, outputs=out2image)
+    
+    btn = gr.Button("Generate mode detailed description.")
+    btn.click(fn=chatgpt, inputs=inpprompt, outputs=outprompt)
+    
+    btn.click(fn=generateimage1, inputs=inpprompt, outputs=out1image)
+    btn.click(fn=generateimage2, inputs=outprompt, outputs=out2image)
     
 
 demo.launch()