Update app.py

app.py

@@ -39,34 +39,25 @@ def fix_json_output(output):
 
 def merge_json_with_amount(model_output, amount):
     """
-    Merges the model's JSON output with the extracted amount.
-    It preserves the values for keys (like "action", "currency", "recipient")
-    that the model provided and only updates the "amount" field.
+    Updates only the 'amount' field in the model's JSON output,
+    leaving all other fields as produced by the model.
     """
     try:
-        # First, try to parse the model output as-is
+        # Attempt to load the model output directly as JSON.
         data = json.loads(model_output)
     except json.JSONDecodeError:
-        # If parsing fails, try to fix common JSON issues
-        fixed_output = fix_json_output(model_output)
-        try:
-            data = json.loads(fixed_output)
-        except json.JSONDecodeError:
-            # If it still fails, start with an empty dictionary
-            data = {}
+        # If parsing fails, just return the model output unmodified.
+        # (You might choose to log an error here.)
+        return model_output
 
-    # Override or add the "amount" field with the extracted value
+    # Replace (or add) the "amount" field using the extracted amount.
     if amount:
         data["amount"] = float(amount) if '.' in amount else int(amount)
-
-    # Ensure that the keys for action, currency, and recipient exist.
-    # Only fill in "unknown" if they are missing or empty.
-    for key in ["action", "currency", "recipient"]:
-        if key not in data or not data[key]:
-            data[key] = "unknown"
-
+    
+    # Dump back to JSON without altering other keys.
     return json.dumps(data, ensure_ascii=False)
 
+
 def generate_command(input_command):
     # Extract the amount from the input
     amount = extract_amount(input_command)