Update app.py

app.py

@@ -881,7 +881,7 @@ If a MCP server requires a password input you will use {pwdList[0]}.
         model= "gpt-5-mini",      #"gpt-5-mini", "o4-mini",
         tools=[{ "type": "web_search" },
                 { "type": "code_interpreter", "container": container_id},    #{'type': 'auto'}},
-                { "type": "image_generation", "quality": "medium", "size": "1024x1024"},
+                { "type": "image_generation", "model": "gpt-image-2", "quality": "medium", "size": "1536x1024"},
             {"type": "function", "name": "get_distance",
               "description": "get calculated straight-line (great-circle) distance between two locations or addresses.",
               "parameters": {
@@ -1487,7 +1487,9 @@ def make_image(prompt, user, pwd):
             msg = ex.message
             return [gr.Image(visible=False, value=None), msg]
 
-        pnum = 0
+        pnum = -1
+        msgs = ["Working on it ...","Making progress ...",
+               "Getting closer ...", "Almost there ..."]
         for event in response:
             if event.type == "image_generation.partial_image":
                 idx = event.partial_image_index
@@ -1497,25 +1499,9 @@ def make_image(prompt, user, pwd):
                 with open(fpath, "wb") as fp:
                     fp.write(image_bytes)
                 pnum += 1
-                yield(gr.Image(value=None, visible=False), f"iteration {pnum}")
-
-        # if len(response.data) == 0:
-        #     msg = "OpenAI returned no image data"
-        #     return [gr.Image(visible=False, value=None), msg]
-        # try:
-        #     image_data = response.data[0].b64_json 
-        #     with Image.open(BytesIO(base64.b64decode(image_data))) as image:
-        #         fpath = dataDir + user + '.png'
-        #         image.save(fpath)
-        #     with open(image_count_path(user), 'at') as fp:
-        #         if quality == 'hd':
-        #             fp.write('hd\n')
-        #         else:
-        #             fp.write('1\n')
+                yield(gr.Image(value=None, visible=False), msgs[pnum])
 
         msg = 'Image created!'
-        # except:
-        #     return [gr.Image(visible=False, value=None), msg]
     else:
         msg = 'Incorrect user name or password'
         return [gr.Image(visible=False, value=None), msg]