Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import re
 import shutil
 import base64
 import tempfile
@@ -42,6 +41,7 @@ def ezgif_convert(media_bytes: bytes, filename: str = "input"):
     files = {"new-image": (filename, media_bytes)}
     r = requests.post("https://s.ezgif.com/upload", files=files, timeout=60)
     r.raise_for_status()
+    import re
     m = re.search(r'name="file" value="([^"]+)"', r.text)
     if not m:
         raise RuntimeError("ezgif upload failed")
@@ -81,16 +81,7 @@ def convert_to_jpeg_bytes(media_bytes: bytes, filename_hint: str = "input"):
 def to_b64_jpeg(img_bytes: bytes):
     return base64.b64encode(img_bytes).decode("utf-8")
 
-def build_prompt(use_auto: bool, custom: str):
-    if custom and custom.strip():
-        return custom.strip()
-    if use_auto:
-        return ("Provide an exhaustive, highly detailed visual description suitable as an image-generation prompt. "
-                "Include objects, materials, textures, colors, lighting, viewpoint, camera lens and settings, "
-                "poses, facial expressions, clothing, background elements, small distinguishing details, mood, and composition.")
-    return "Provide a detailed description of this image."
-
-def generate_stream(image_src: str, use_auto: bool, custom_prompt: str, alt_key: str):
+def generate_stream(image_src: str, custom_prompt: str, alt_key: str):
     try:
         raw = fetch_bytes(image_src)
         jpg = convert_to_jpeg_bytes(raw, filename_hint=os.path.basename(image_src) or "input")
@@ -99,12 +90,13 @@ def generate_stream(image_src: str, use_auto: bool, custom_prompt: str, alt_key:
         return
 
     b64 = to_b64_jpeg(jpg)
-    prompt = build_prompt(use_auto, custom_prompt)
+    prompt_text = (custom_prompt.strip() if custom_prompt and custom_prompt.strip() else
+                   "Provide a detailed, neutral, clinical-style description of the image focusing on observable non-sexual features, hygiene, skin condition, posture, and general anatomy. Keep language professional.")
     model = "pixtral-12b-2409"
     messages = [{
         "role": "user",
         "content": [
-            {"type": "text", "text": prompt},
+            {"type": "text", "text": prompt_text},
             {"type": "image_url", "image_url": f"data:image/jpeg;base64,{b64}"}
         ],
         "stream": False
@@ -121,18 +113,17 @@ def generate_stream(image_src: str, use_auto: bool, custom_prompt: str, alt_key:
         yield f"Model error: {e}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("Image To Flux Prompt")
+    gr.Markdown("Image/Video to Clinical Description (custom prompt optional)")
 
     with gr.Row():
         with gr.Column(scale=1):
             alt_key = gr.Textbox(label="Alternate Mistral API Key (optional)", type="password")
-            url_input = gr.Textbox(label="Image URL", placeholder="https://...")
-            custom = gr.Textbox(label="Custom prompt (optional)", lines=4, placeholder="Leave blank to use auto prompt")
-            auto_toggle = gr.Checkbox(label="Use detailed auto prompt", value=True)
-            preview = gr.Image(label="Preview", type="pil")
+            url_input = gr.Textbox(label="Image/Video URL", placeholder="https://...")
+            custom = gr.Textbox(label="Custom prompt (optional)", lines=4, placeholder="Enter custom prompt to override default")
             submit = gr.Button("Submit")
+            preview = gr.Image(label="Preview", type="pil")
         with gr.Column(scale=1):
-            out = gr.Textbox(label="Generated Detailed Description", lines=20)
+            out = gr.Textbox(label="Generated Detailed Description", lines=25)
 
     def load_preview(url):
         if not url:
@@ -145,13 +136,13 @@ with gr.Blocks() as demo:
         except Exception:
             return None
 
-    def start_gen(url, use_auto, custom_p, alt_k):
+    def start_gen(url, custom_p, alt_k):
         if not url:
             return "No URL provided."
-        for chunk in generate_stream(url, use_auto, custom_p, alt_k):
+        for chunk in generate_stream(url, custom_p, alt_k):
             yield chunk
 
     url_input.change(fn=load_preview, inputs=[url_input], outputs=[preview])
-    submit.click(fn=start_gen, inputs=[url_input, auto_toggle, custom, alt_key], outputs=[out])
+    submit.click(fn=start_gen, inputs=[url_input, custom, alt_key], outputs=[out])
 
 demo.launch()