Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import pipeline
+import json
 
 # Define the JSON structure
 predefined_json = {
@@ -19,17 +20,18 @@ model = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_lengt
 
 def convert_text_to_json(input_text):
     # Generate JSON using the LLM
-    prompt = f"Extract the following fields into JSON format: {list(predefined_json.keys())}\nInput: {input_text}"
+    prompt = f"Extrae los campos incluidos en el siguiente formato JSON: {list(predefined_json.keys())}\nInput: {input_text}"
     response = model(prompt)[0]["generated_text"]
 
-    # Attempt to parse the response into JSON
+    # Attempt to parse the response into a JSON object
     try:
-        generated_json = eval(response)  # Unsafe for prod, use `json.loads()` for proper JSON
-        missing_fields = [key for key in predefined_json if not generated_json.get(key)]
-        missing_message = f"Missing fields: {', '.join(missing_fields)}" if missing_fields else "All fields are complete."
-    except Exception as e:
-        generated_json = {}
-        missing_message = f"Error parsing JSON: {e}"
+        generated_json = json.loads(response)  # Safer than eval()
+    except json.JSONDecodeError:
+        return {}, "Error: El modelo no retornó un formato JSON válido."
+
+    # Check for missing fields
+    missing_fields = [key for key in predefined_json if key not in generated_json or not generated_json[key]]
+    missing_message = f"Campos faltantes: {', '.join(missing_fields)}" if missing_fields else "Todos los campos están completos."
 
     return generated_json, missing_message
 
@@ -38,8 +40,8 @@ interface = gr.Interface(
     fn=convert_text_to_json,
     inputs="text",
     outputs=["json", "text"],
-    title="Text to JSON Converter",
-    description="Enter text to extract information into predefined JSON format."
+    title="Convertidor Texto en JSON",
+    description="Ingrese el texto para extraer información en un formato JSON predefinido."
 )
 
-interface.launch()
+interface.launch()