Update app.py

app.py

@@ -1,7 +1,5 @@
-import os, base64, io
 import gradio as gr
 from openai import OpenAI
-from PIL import Image
 
 # ------------------------------------------------------------
 #  CLIENT SETUP
@@ -13,7 +11,7 @@ def get_client(key: str):
     return OpenAI(api_key=key)
 
 # ------------------------------------------------------------
-#  TEXT CHAT (GPT-5)
+#  GPT-5 CHAT
 # ------------------------------------------------------------
 def chat_response(api_key, prompt, history):
     client = get_client(api_key)
@@ -25,7 +23,7 @@ def chat_response(api_key, prompt, history):
 
     try:
         stream = client.chat.completions.create(
-            model="gpt-5",  # fallback to gpt-4o if your account doesn’t yet support 5
+            model="gpt-5",
             messages=messages,
             stream=True,
         )
@@ -38,38 +36,33 @@ def chat_response(api_key, prompt, history):
         yield f"[Error] {e}"
 
 # ------------------------------------------------------------
-#  IMAGE GENERATION (DALL·E 3 / gpt-image-1)
+#  CODE EXPLAINER / REFACTOR
 # ------------------------------------------------------------
-def generate_image(api_key, prompt, size):
+def analyze_code(api_key, code_snippet, mode):
     client = get_client(api_key)
-    try:
-        response = client.images.generate(
-            model="gpt-image-1",
-            prompt=prompt,
-            size=size,
-            n=1,
-        )
+    if not code_snippet.strip():
+        raise gr.Error("Paste some code first.")
 
-        # Defensive extraction: OpenAI may return base64 or URL depending on account
-        data = response.data[0]
-        if hasattr(data, "b64_json") and data.b64_json:
-            img_bytes = base64.b64decode(data.b64_json)
-            return Image.open(io.BytesIO(img_bytes))
-        elif hasattr(data, "url") and data.url:
-            # fallback if only URL provided
-            import requests
-            img_bytes = requests.get(data.url).content
-            return Image.open(io.BytesIO(img_bytes))
-        else:
-            raise ValueError("No valid image data returned from API.")
+    prompt = (
+        "Explain this code clearly and suggest improvements:\n\n" + code_snippet
+        if mode == "Explain & Improve"
+        else "Refactor this code for clarity, efficiency, and style:\n\n" + code_snippet
+    )
 
+    try:
+        completion = client.chat.completions.create(
+            model="gpt-5",
+            messages=[{"role": "system", "content": "You are an expert software engineer."},
+                      {"role": "user", "content": prompt}],
+        )
+        return completion.choices[0].message.content
     except Exception as e:
-        raise gr.Error(f"Image generation failed: {e}")
+        raise gr.Error(f"Code analysis failed: {e}")
 
 # ------------------------------------------------------------
 #  INTERFACE
 # ------------------------------------------------------------
-with gr.Blocks(title="ZEN GPT-5 + DALL·E 3") as demo:
+with gr.Blocks(title="ZEN GPT-5 Production SDK") as demo:
     gr.Markdown("### 🔐 Enter your OpenAI API key (not stored)")
     key = gr.Textbox(placeholder="sk-...", type="password", label="OpenAI API Key")
 
@@ -81,24 +74,28 @@ with gr.Blocks(title="ZEN GPT-5 + DALL·E 3") as demo:
 
         def chat_submit(k, m, h):
             h = h or []
-            h.append({"role": "user", "content": m})
-            for r in chat_response(k, m, [(x["content"], y["content"]) for x, y in zip(h, h)]):
-                h.append({"role": "assistant", "content": r})
+            h.append([m, ""])
+            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"
-        )
-        gen = gr.Button("Generate Image")
-        img_out = gr.Image(label="Result", height=512)
-        gen.click(generate_image, [key, prompt, size], img_out)
+    with gr.Tab("💻 Code Assistant"):
+        gr.Markdown("Use GPT-5 to **explain or refactor** any code.")
+        code_input = gr.Code(label="Paste your code here", language="python", lines=18)
+        mode = gr.Radio(["Explain & Improve", "Refactor Only"], value="Explain & Improve", label="Mode")
+        run_btn = gr.Button("Analyze Code", variant="primary")
+        result = gr.Textbox(label="Result", lines=18)
+        run_btn.click(analyze_code, [key, code_input, mode], result)
+
+    # footer / watermark
+    gr.HTML(
+        "<div style='text-align:right; font-size:12px; opacity:0.5; margin-top:12px;'>"
+        "Module 3 – ZEN SDK Production"
+        "</div>"
+    )
 
 demo.launch()