Update app.py

app.py

@@ -2,13 +2,11 @@ import os
 import gradio as gr
 from cloudflare import Cloudflare
 #from typing import List, Union
-
-client = Cloudflare(api_token=os.environ['CF_API_TOKEN'])
-
+API_TOKEN=os.environ.get('CF_API_TOKEN')
 # set account id and model name
-ACCOUNT_ID = os.environ['CF_ACCOUNT_ID']
-IMAGE_MODEL = os.environ['CF_IMAGE_MODEL']
-CHAT_MODEL = os.environ['CF_CHAT_MODEL']
+ACCOUNT_ID = os.environ.get('CF_ACCOUNT_ID')
+IMAGE_MODEL = os.environ.get('CF_IMAGE_MODEL')
+CHAT_MODEL = os.environ.get('CF_CHAT_MODEL')
 
 PRESET_PROMPT = """
 你是一个基于Flux.1模型的提示词生成机器人。根据用户的需求，自动生成符合Flux.1格式的绘画提示词。虽然你可以参考提供的模板来学习提示词结构和规律，但你必须具备灵活性来应对各种不同需求。最终输出应仅限提示词，无需任何其他解释或信息。你的回答必须全部使用英语进行回复我！
@@ -87,6 +85,8 @@ PRESET_PROMPT = """
 **你的输出**：A futuristic cityscape at night, in a cyberpunk style, with neon lights reflecting off wet streets, towering skyscrapers, and a glowing, high-tech atmosphere. Dark shadows contrast with vibrant neon signs, creating a dramatic, dystopian mood`
 """
 
+client = Cloudflare(api_token=API_TOKEN)
+
 def generate_image(prompt):
     try:
         image = client.workers.ai.with_raw_response.run(
@@ -136,8 +136,8 @@ with gr.Blocks(theme="ocean") as demo:
     gr.HTML("<h1><center>ImagenX</center></h1>")
     gr.HTML("<p><center>Powered By Cloudflare + Gradio</center></p>")
     with gr.Row():
-        gen_text = gr.Markdown(label="Optimized Text")
-        gen_img = gr.Image(type="pil", label='Generated Image')
+        gen_text = gr.Markdown(label="Optimized Text", height=600)
+        gen_img = gr.Image(type="pil", label='Generated Image', height=600)
     with gr.Row():
         prompt = gr.Textbox(label='Enter Your Prompt', placeholder="Enter prompt...", scale=6)
         sendBtn = gr.Button(variant='primary')