Update main.py

main.py

@@ -145,8 +145,8 @@ def chat(product_name: str = Form(...)):
             model="gpt-4o-mini",
             messages=[{"role": "user", "content": prompt.format(product_name=product_name)}],
         )
-        result = completion.choices[0].message
-        logger.info(f"OpenAI response: {result.content}")
+        result = completion.choices[0].message.content
+        logger.info(f"OpenAI response: {result}")
         return result
     except Exception as e:
         logger.error(f"Error in OpenAI chat: {str(e)}")
@@ -205,10 +205,9 @@ async def segment_image(
 
 @app.post("/segment/langsam")
 async def segment_image(image_url: str = Form(...), text_prompt: str = Form(...)):
+    # process text prompt using openai chat
+    text_prompt = chat(text_prompt)
     try:
-        # process text prompt using openai chat
-        text_prompt = chat(text_prompt.content)
-
         logger.info(f"Starting LangSAM segmentation for image URL: {image_url} with prompt: {text_prompt}")
         image_pil = load_image_from_url(image_url)
         image_array = np.array(image_pil)