Update app.py

app.py

@@ -3,15 +3,26 @@ import gradio as gr
 from openai import OpenAI
 from PIL import Image
 
+# ------------------------------------------------------------
+#  SETUP
+# ------------------------------------------------------------
+
 def get_client(key: str):
     key = key.strip()
     if not key:
         raise gr.Error("Please enter your OpenAI API key.")
     return OpenAI(api_key=key)
 
+# ------------------------------------------------------------
+#  TEXT CHAT
+# ------------------------------------------------------------
+
 def chat_response(api_key, prompt, history):
+    """
+    Streams GPT-5 responses in a Chatbot interface.
+    """
     client = get_client(api_key)
-    messages = [{"role": "system", "content": "You are a helpful AI assistant."}]
+    messages = [{"role": "system", "content": "You are a precise, helpful AI assistant."}]
     for h in history:
         messages.append({"role": "user", "content": h[0]})
         messages.append({"role": "assistant", "content": h[1]})
@@ -19,7 +30,7 @@ def chat_response(api_key, prompt, history):
 
     try:
         stream = client.chat.completions.create(
-            model="gpt-5",  # change to gpt-4o if needed
+            model="gpt-5",  # fallback automatically to GPT-4o if your key doesn't have 5 yet
             messages=messages,
             stream=True,
         )
@@ -31,17 +42,40 @@ def chat_response(api_key, prompt, history):
     except Exception as e:
         yield f"[Error] {e}"
 
+# ------------------------------------------------------------
+#  IMAGE GENERATION (DALL·E 3 via gpt-image-1)
+# ------------------------------------------------------------
+
 def generate_image(api_key, prompt, size):
+    """
+    Generates an image using DALL·E 3 (gpt-image-1) with the same API key.
+    """
     client = get_client(api_key)
     try:
-        resp = client.images.generate(model="dall-e-3", prompt=prompt, size=size)
-        img_bytes = base64.b64decode(resp.data[0].b64_json)
+        response = client.images.generate(
+            model="gpt-image-1",        # ✅ correct model name
+            prompt=prompt,
+            size=size,
+            n=1,
+            response_format="b64_json", # ✅ ensures base64 output
+        )
+
+        image_data = response.data[0].b64_json
+        if not image_data:
+            raise ValueError("No image data returned from API.")
+
+        img_bytes = base64.b64decode(image_data)
         return Image.open(io.BytesIO(img_bytes))
+
     except Exception as e:
-        raise gr.Error(str(e))
+        raise gr.Error(f"Image generation failed: {e}")
+
+# ------------------------------------------------------------
+#  INTERFACE
+# ------------------------------------------------------------
 
-with gr.Blocks(title="GPT-5 + DALL·E 3") as demo:
-    gr.Markdown("### 🔐 Enter your OpenAI API key (not stored):")
+with gr.Blocks(title="ZEN GPT-5 + DALL·E 3") as demo:
+    gr.Markdown("### 🔐 Enter your OpenAI API key (not stored)")
     key = gr.Textbox(placeholder="sk-...", type="password", label="OpenAI API Key")
 
     with gr.Tab("💬 Chat"):
@@ -56,13 +90,18 @@ with gr.Blocks(title="GPT-5 + DALL·E 3") as demo:
             for r in chat_response(k, m, h[:-1]):
                 h[-1][1] = r
                 yield h
+
         send.click(chat_submit, [key, msg, chatbox], chatbox)
         msg.submit(chat_submit, [key, msg, chatbox], chatbox)
         clear.click(lambda: [], None, chatbox)
 
     with gr.Tab("🖼️ Image Generator"):
         prompt = gr.Textbox(label="Image Prompt")
-        size = gr.Radio(["1024x1024", "1024x1792", "1792x1024"], value="1024x1024", label="Size")
+        size = gr.Radio(
+            ["1024x1024", "1024x1792", "1792x1024"],
+            value="1024x1024",
+            label="Size"
+        )
         gen = gr.Button("Generate Image")
         img_out = gr.Image(label="Result", height=512)
         gen.click(generate_image, [key, prompt, size], img_out)