Update app.py

app.py

@@ -116,19 +116,47 @@ def polish_prompt_hf(prompt, img_list):
             api_key=api_key,
         )
 
+        image_url = None
+        if img is not None:
+            # If img is a PIL Image
+            if hasattr(img, 'save'):  # Check if it's a PIL Image
+                buffered = BytesIO()
+                img.save(buffered, format="PNG")
+                img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+                image_url = f"data:image/png;base64,{img_base64}"
+            # If img is already a file path (string)
+            elif isinstance(img, str):
+                with open(img, "rb") as image_file:
+                    img_base64 = base64.b64encode(image_file.read()).decode('utf-8')
+                image_url = f"data:image/png;base64,{img_base64}"
+            else:
+                print(f"Warning: Unexpected image type: {type(img)}")
+                return original_prompt
+
         # Format the messages for the chat completions API
-        sys_promot = "you are a helpful assistant, you should provide useful answers to users."
         messages = [
-            {"role": "system", "content": sys_promot},
-            {"role": "user", "content": []}]
-        for img in img_list:
-            messages[1]["content"].append(
-                {"image": f"data:image/png;base64,{encode_image(img)}"})
-        messages[1]["content"].append({"text": f"{prompt}"})
+            {"role": "system", "content": system_prompt},
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": original_prompt
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": image_url
+                        }
+                    }
+                ]
+            }
+        ]
+
 
         # Call the API
         completion = client.chat.completions.create(
-            model="Qwen/Qwen3-235B-A22B-Instruct-2507",
+            model="Qwen/Qwen2.5-VL-72B-Instruct",
             messages=messages,
         )
         
@@ -149,11 +177,6 @@ def polish_prompt_hf(prompt, img_list):
             
         polished_prompt = polished_prompt.strip().replace("\n", " ")
         return polished_prompt
-        
-    except Exception as e:
-        print(f"Error during API call to Hugging Face: {e}")
-        # Fallback to original prompt if enhancement fails
-        return prompt