Update app.py

app.py

@@ -3,7 +3,8 @@ import json
 import gradio as gr
 import groq
 from difflib import Differ
-from json2kadi import transform_value, transform_json2kadi
+from json2kadi import transform_value, my_json_to_kadi
+from kadi_apy.lib.conversion import json_to_kadi
 
 # Set api key of Groq
 api_key = os.getenv("GROQ_API")
@@ -215,7 +216,11 @@ def diff_texts(text1, text2):
 
 def transform_json_to_kadi_schema(input_json_str):
     input_json = json.loads(input_json_str)
-    output_json = transform_json2kadi(input_json)
+    try:
+        output_json = my_json_to_kadi(input_json)
+    except Exception:
+        # fallback to json_to_kadi from kadi_apy
+        output_json = json_to_kadi(input_json)
 
     return json.dumps(output_json, indent=2)