Update app.py

app.py

@@ -164,12 +164,31 @@ def llm_chat_response(text: str, image_url: Optional[str] = None) -> str:
         ]
 
         logger.info("Sending request to model...")
+        logger.info(f"Request payload: {messages}")
+        
         completion = client.chat.completions.create(
             model="meta-llama/Llama-3.2-11B-Vision-Instruct",
             messages=messages,
             max_tokens=500
         )
-        return completion.choices[0].message['content']
+        
+        logger.info(f"Response received: {completion}")
+        
+        # Check the structure of the response and extract content
+        if hasattr(completion, 'choices') and len(completion.choices) > 0:
+            message = completion.choices[0].message
+            # Handle different response formats
+            if isinstance(message, dict) and 'content' in message:
+                return message['content']
+            elif hasattr(message, 'content'):
+                return message.content
+            else:
+                logger.error(f"Unexpected message format: {message}")
+                return str(message)
+        else:
+            logger.error(f"Unexpected completion format: {completion}")
+            return str(completion)
+            
     except Exception as e:
         logger.error(f"Error in llm_chat_response: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))