backup

app.py

@@ -41,54 +41,18 @@ model = PeftModel.from_pretrained(model, lora_weights_path)
 model.tie_weights()
 
 
-def parse_json_response(json_str):
-    if not json_str:
-        print("DEBUG: Empty JSON string received")
-        return None
-
+def describe_image_in_JSON(json_string):
     try:
-        # Debug original input
-        print("DEBUG: Raw JSON input:", json_str)
-
-        # Handle potential JSON string escaping
-        json_str = json_str.strip()
-        if json_str.startswith('"') and json_str.endswith('"'):
-            print("DEBUG: Removing outer quotes")
-            json_str = json_str[1:-1]
-
-        # First parse - handles the outer JSON encoding
-        first_parse = json.loads(json_str)
-        print("DEBUG: First parse result type:", type(first_parse))
-
-        # Second parse - if the result is still a string, parse again
-        json_object = (
-            json.loads(first_parse) if isinstance(first_parse, str) else first_parse
-        )
-        print("DEBUG: Final parsed object type:", type(json_object))
-
-        # Validate expected keys
-        required_keys = [
-            "description",
-            "scene_description",
-            "character_list",
-            "object_list",
-        ]
-        missing_keys = [key for key in required_keys if key not in json_object]
-        if missing_keys:
-            print("DEBUG: Missing required keys:", missing_keys)
-            return None
+        # First JSON decode
+        first_decode = json.loads(json_string)
+
+        # Second JSON decode - parse the actual data
+        final_data = json.loads(first_decode)
+
+        return final_data
 
-        return json_object
     except json.JSONDecodeError as e:
-        print(f"DEBUG: JSON parsing error: {e}")
-        print(f"DEBUG: Error occurred at position {e.pos}")
-        print(
-            f"DEBUG: Problematic document snippet: {json_str[max(0, e.pos-50):e.pos+50]}"
-        )
-        return None
-    except Exception as e:
-        print(f"DEBUG: Unexpected error during JSON parsing: {e}")
-        return None
+        return f"Error parsing JSON: {str(e)}"
 
 
 def create_color_palette_image(colors):
@@ -181,7 +145,7 @@ def inference(image):
             "Error",
         ]
 
-    parsed_json = parse_json_response(json_str)
+    parsed_json = describe_image_in_JSON(json_str)
     if parsed_json:
         # Create color palette visualization
         colors = parsed_json.get("color_palette", [])