Fix JSON array generation - add explicit array requirements and improve JSON parsing

gradio_app.py

@@ -152,9 +152,16 @@ def create_json_prompt(message, template_type):
 
 {template["instruction"]}
 
+IMPORTANT: You must respond with EXACTLY this format:
+
 {template["schema"]}
 
-Ensure the response is valid JSON that can be parsed. Do not include any text outside the JSON structure.
+Requirements:
+- Start with [ and end with ]
+- Include multiple items in the array
+- Use valid JSON syntax
+- Do not include any text before or after the JSON
+- Complete the entire JSON structure
 
 <|eot_id|><|start_header_id|>assistant<|end_header_id|>
 
@@ -163,16 +170,29 @@ Ensure the response is valid JSON that can be parsed. Do not include any text ou
 def prettify_json_response(response_text):
     """Try to extract and prettify JSON from response"""
     try:
+        # Clean the response first
+        cleaned = response_text.strip()
+        
+        # Try to parse the entire response as JSON first
+        try:
+            parsed_json = json.loads(cleaned)
+            return json.dumps(parsed_json, indent=2, ensure_ascii=False)
+        except json.JSONDecodeError:
+            pass
+        
         # Try to find JSON in the response - look for both objects and arrays
+        # Use non-greedy matching and better patterns
         json_patterns = [
-            r'\[.*\]',  # Array pattern
-            r'\{.*\}'   # Object pattern
+            r'\[[\s\S]*?\](?=\s*$)',  # Array pattern - non-greedy, end of string
+            r'\{[\s\S]*?\}(?=\s*$)',  # Object pattern - non-greedy, end of string
+            r'\[[\s\S]*\]',          # Array pattern - greedy fallback
+            r'\{[\s\S]*\}'           # Object pattern - greedy fallback
         ]
         
         for pattern in json_patterns:
-            json_match = re.search(pattern, response_text, re.DOTALL)
+            json_match = re.search(pattern, cleaned, re.MULTILINE)
             if json_match:
-                json_str = json_match.group()
+                json_str = json_match.group().strip()
                 try:
                     parsed_json = json.loads(json_str)
                     return json.dumps(parsed_json, indent=2, ensure_ascii=False)