Update app.py

app.py

@@ -475,7 +475,6 @@ from transformers import AutoModelForCausalLM, AutoProcessor, AutoTokenizer
 import tempfile
 import os
 import base64
-from openai import OpenAI
 
 def encode_image(image_path):
     try:
@@ -486,53 +485,6 @@ def encode_image(image_path):
         print(f"Error in encode_image: {str(e)}")
         raise
 
-client = OpenAI(
-    api_key=os.getenv("DASHSCOPE_API_KEY", 'sk-5d71cf15539f46ef9ea9283a821f7ee7'),
-    base_url="https://dashscope-intl.aliyuncs.com/compatible-mode/v1"
-)
-
-def inference_with_api(image_path, prompt, sys_prompt="You are a helpful visual analysis assistant that specializes in determining how products and people should be positioned on a canvas for optimal visual presentation.", model_id="qwen2.5-vl-72b-instruct", min_pixels=512*28*28, max_pixels=2048*28*28):
-    try:
-        base64_image = encode_image(image_path)
-        messages = [
-            {
-                "role": "system",
-                "content": [{"type": "text", "text": sys_prompt}]
-            },
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "image_url",
-                        "min_pixels": min_pixels,
-                        "max_pixels": max_pixels,
-                        "image_url": {"url": f"data:image/jpeg;base64,{base64_image}"},
-                    },
-                    {"type": "text", "text": prompt},
-                ],
-            }
-        ]
-        retries = 3
-        for attempt in range(retries):
-            try:
-                completion = client.chat.completions.create(
-                    model=model_id,
-                    messages=messages,
-                    timeout=30  # Meningkatkan timeout dari 15 ke 30 detik
-                )
-                return completion.choices[0].message.content
-            except Exception as inner_e:
-                # If the error message contains "Connection error", retry
-                if "Connection error" in str(inner_e):
-                    print(f"Connection error on attempt {attempt+1}: {inner_e}. Retrying in 2 seconds...")
-                    time.sleep(2)
-                else:
-                    raise
-        raise Exception("Failed to complete API call after multiple retries due to connection errors.")
-    except Exception as e:
-        print(f"Error in inference_with_api: {str(e)}")
-        raise
-
 def classify_image(image_path, unique_items):
     try:
         image = Image.open(image_path).convert("RGB")