Update app.py

app.py

@@ -24,6 +24,7 @@ def merge_json_with_amount(model_output, amount):
         data = json.loads(model_output)  # Try parsing the model output
     except json.JSONDecodeError:
         return model_output  # Return raw model output if it's not valid JSON
+        data["amount"] = amount
 
    # if amount:
    #     try: