udpated message response

main.py

@@ -89,11 +89,11 @@ def run_ai_inference(diff: str) -> str:
     messages = [
         {
             "role": "system",
-            "content": "You are a code reviewer. Analyze the diff and return a JSON array of comments with file_path, line_number, and comment_text fields. Be concise."
+            "content": "You are a code reviewer. Analyze the diff and respond with ONLY a JSON array. No other text, no markdown, no explanations. Just the JSON array with file_path, line_number, and comment_text fields."
         },
         {
             "role": "user",
-            "content": f"Review this diff:\n{diff[:1000]}"  # Limit diff size for faster processing
+            "content": f"Analyze this specific diff and provide review comments:\n{diff[:800]}"  # Slightly reduced for faster processing
         }
     ]
 
@@ -120,22 +120,48 @@ def parse_ai_response(response_text: str) -> list[ReviewComment]:
     """
     print(f"Raw AI Response:\n---\n{response_text}\n---")
     
+    # Try to find JSON array, handling both direct JSON and markdown-wrapped JSON
     json_match = re.search(r'\[.*?\]', response_text, re.DOTALL)
     if not json_match:
-        print("Warning: Could not find a JSON array in the AI response.")
-        # Return a simple fallback comment
-        return [ReviewComment(
-            file_path="unknown", 
-            line_number=1, 
-            comment_text="AI review completed - no specific issues found."
-        )]
-
-    json_string = json_match.group(0)
+        # Try to find JSON inside markdown code blocks
+        markdown_match = re.search(r'```json\s*(\[.*?\])\s*```', response_text, re.DOTALL)
+        if markdown_match:
+            json_match = markdown_match
+            json_string = markdown_match.group(1)
+        else:
+            print("Warning: Could not find a JSON array in the AI response.")
+            # Return a simple fallback comment
+            return [ReviewComment(
+                file_path="unknown", 
+                line_number=1, 
+                comment_text="AI review completed - no specific issues found."
+            )]
+    else:
+        json_string = json_match.group(0)
     
     try:
         comments_data = json.loads(json_string)
-        validated_comments = [ReviewComment(**item) for item in comments_data]
-        return validated_comments
+        if not isinstance(comments_data, list):
+            print("Warning: AI response is not a JSON array")
+            return [ReviewComment(
+                file_path="unknown", 
+                line_number=1, 
+                comment_text="AI review completed - format issue."
+            )]
+            
+        validated_comments = []
+        for item in comments_data:
+            try:
+                validated_comments.append(ReviewComment(**item))
+            except (TypeError, ValueError) as e:
+                print(f"Skipping invalid comment: {item}, error: {e}")
+                
+        return validated_comments if validated_comments else [ReviewComment(
+            file_path="unknown", 
+            line_number=1, 
+            comment_text="AI review completed - no valid comments found."
+        )]
+        
     except (json.JSONDecodeError, TypeError, KeyError) as e:
         print(f"Error parsing JSON from AI response: {e}")
         print(f"Invalid JSON string: {json_string}")