Update app.py

app.py

@@ -233,11 +233,51 @@ def build_messages_for_text(prompt: str, extra_text: str):
     return [{"role": "system", "content": SYSTEM_INSTRUCTION}, {"role": "user", "content": f"{prompt}\n\n{extra_text}"}]
 
 
+# New helper: normalize messages so content is always a plain string
+def normalize_messages(messages):
+    out = []
+    for m in messages:
+        if not isinstance(m, dict):
+            out.append(m)
+            continue
+        c = m.get("content")
+        if isinstance(c, list):
+            parts = []
+            for item in c:
+                if isinstance(item, str):
+                    parts.append(item)
+                elif isinstance(item, dict):
+                    typ = item.get("type")
+                    if typ == "text" and item.get("text"):
+                        parts.append(item["text"])
+                    elif typ == "image_url" and item.get("image_url"):
+                        parts.append(item["image_url"])
+                    elif typ == "image_base64" and item.get("image_base64"):
+                        # convert to data URL to satisfy string requirement
+                        parts.append("data:image/jpeg;base64," + item["image_base64"])
+                    else:
+                        parts.append(item.get("text") or item.get("image_url") or item.get("image_base64") or "")
+                else:
+                    parts.append(str(item))
+            newc = "\n\n".join(p for p in parts if p).strip()
+            nm = m.copy()
+            nm["content"] = newc
+            out.append(nm)
+        elif not isinstance(c, str):
+            nm = m.copy()
+            nm["content"] = str(c or "")
+            out.append(nm)
+        else:
+            out.append(m)
+    return out
+
+
 def stream_and_collect(client, model, messages, parts: list):
     try:
+        norm_msgs = normalize_messages(messages)
         stream_gen = None
         try:
-            stream_gen = client.chat.stream(model=model, messages=messages)
+            stream_gen = client.chat.stream(model=model, messages=norm_msgs)
         except Exception:
             stream_gen = None
         if stream_gen:
@@ -249,7 +289,7 @@ def stream_and_collect(client, model, messages, parts: list):
                     continue
                 parts.append(d)
             return
-        res = client.chat.complete(model=model, messages=messages, stream=False)
+        res = client.chat.complete(model=model, messages=norm_msgs, stream=False)
         try:
             choices = getattr(res, "choices", None) or res.get("choices", [])
         except Exception:
@@ -404,6 +444,7 @@ with gr.Blocks(title="Flux", css=css) as demo:
         with gr.Column(scale=2):
             final_text = gr.Markdown(value="")
 
+    # Ensure preview outputs get None when not applicable so Gradio hides them
     url_input.change(fn=load_preview, inputs=[url_input], outputs=[preview_image, preview_video, gr.Textbox(visible=False)])
     submit.click(fn=generate_final_text, inputs=[url_input, custom_prompt, api_key], outputs=[final_text])
     demo.queue()