Update app.py

app.py

@@ -136,8 +136,8 @@ def llm_chat_response(text: str, image_url: Optional[str] = None) -> str:
             api_key=HF_TOKEN
         )
         
-        # Build the messages payload dynamically based on whether an image URL is provided.
-        # If only text is provided, add an instruction for a one-line description.
+        # Build the messages payload dynamically.
+        # If only text is provided, we append a default instruction.
         message_content = [{
             "type": "text",
             "text": text + ("" if image_url else " describe in one line only")
@@ -160,7 +160,41 @@ def llm_chat_response(text: str, image_url: Optional[str] = None) -> str:
             messages=messages,
             max_tokens=500
         )
-        return completion.choices[0].message['content']
+        
+        # Debug log the raw response for troubleshooting.
+        logger.info(f"Raw model response: {completion}")
+        
+        # Ensure we have a valid response.
+        if not completion.choices or len(completion.choices) == 0:
+            logger.error("No choices returned from model.")
+            raise HTTPException(status_code=500, detail="Model returned no choices.")
+        
+        # Extract the message; use dict get to avoid NoneType errors.
+        response_message = None
+        # Some responses may be dicts or objects; try both approaches.
+        choice = completion.choices[0]
+        if hasattr(choice, "message"):
+            response_message = choice.message
+        elif isinstance(choice, dict):
+            response_message = choice.get("message")
+        
+        if not response_message:
+            logger.error(f"Response message is empty: {choice}")
+            raise HTTPException(status_code=500, detail="Model response did not include a message.")
+        
+        # Extract the content from the message.
+        content = None
+        if isinstance(response_message, dict):
+            content = response_message.get("content")
+        # If for some reason it's not a dict, try attribute access.
+        if content is None and hasattr(response_message, "content"):
+            content = response_message.content
+        
+        if not content:
+            logger.error(f"Message content is missing: {response_message}")
+            raise HTTPException(status_code=500, detail="Model message did not include content.")
+        
+        return content
 
     except Exception as e:
         logger.error(f"Error in llm_chat_response: {str(e)}")
@@ -198,3 +232,4 @@ async def method_not_allowed_handler(request, exc):
         status_code=405,
         content={"error": "Method not allowed. Please check the API documentation."}
     )
+