Update app.py

app.py

@@ -19,17 +19,35 @@ if torch.cuda.is_available():
     except:
         pass
 
+def fix_json_output(output):
+    """
+    Fixes common JSON formatting issues in the model's output.
+    """
+    # Remove trailing commas before closing braces/brackets
+    output = re.sub(r',\s*([}\]])', r'\1', output)
+    # Fix missing or extra quotes around keys
+    output = re.sub(r'([{,])\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*:', r'\1"\2":', output)
+    # Fix missing or extra quotes around string values
+    output = re.sub(r':\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*([,}])', r':"\1"\2', output)
+    return output
+
 def correct_amount_format(output):
-    # Attempt to parse as JSON and correct amounts
+    """
+    Corrects amount formatting in the JSON output.
+    """
     try:
+        # Fix JSON formatting issues
+        output = fix_json_output(output)
         data = json.loads(output)
         
         def correct_value(value):
             if isinstance(value, str):
-                # Remove commas used as thousand separators
-                value = re.sub(r',(?=\d{3})', '', value)
-                # Replace the first comma with a period (decimal)
-                value = value.replace(',', '.', 1)
+                # Fix amounts with multiple decimal points (e.g., 3140.98.0 → 3140.98)
+                if re.match(r'^\d+\.\d+\.\d+$', value):
+                    value = value.split('.')[0] + '.' + value.split('.')[1]
+                # Remove trailing .0 if it's not part of the original amount
+                if re.match(r'^\d+\.0$', value):
+                    value = value.split('.')[0]
             return value
         
         # Correct each value in the JSON data