Update handler.py

handler.py

@@ -15,11 +15,75 @@ class EndpointHandler:
         # Prompt personalizado para guiar al modelo
         input_text = (
             """
-            Genera un JSON válido en el siguiente formato preentrenado:
-            {\"values\": [{\"id\": \"firstName\", \"value\": \"STRING\"},{\"id\": \"lastName\", \"value\": \"STRING\"},{\"id\": \"jobTitle\", \"value\": \"STRING\"},{\"id\": \"adress\", \"value\": [{\"id\": \"[MOBILE-WORK-PERSONAL-MAIN-OTHER]\", \"value\": \"STRING\"}]},{\"id\": \"email\", \"value\": [{\"id\": \"[MOBILE-WORK-PERSONAL-MAIN-OTHER]\", \"value\": \"STRING\"}]},{\"id\": \"phone\", \"value\": [{\"id\": \"[MOBILE-WORK-PERSONAL-MAIN-OTHER]\", \"value\": \"STRING (ONLY NUMBERS)\"}]},{\"id\": \"notes\", \"value\": \"STRING\"},{\"id\": \"roleFunction\", \"value\": \"[BUYER-SELLER-SUPPLIER-PARTNER-COLLABORATOR-PROVIDER-CUSTOMER]\"}]}
-            Procesa el siguiente texto: """
-            + f"\"{data['inputs']}\""
-        )
+            Por favor, genera un JSON válido basado en las siguientes especificaciones:
+
+        Formato esperado:
+        {{
+            "values": [
+                {{
+                    "id": "firstName",
+                    "value": "STRING"
+                }},
+                {{
+                    "id": "lastName",
+                    "value": "STRING"
+                }},
+                {{
+                    "id": "jobTitle",
+                    "value": "STRING"
+                }},
+                {{
+                    "id": "adress",
+                    "value": [
+                        {{
+                            "id": "[MOBILE-WORK-PERSONAL-MAIN-OTHER]",
+                            "value": "STRING"
+                        }}
+                    ]
+                }},
+                {{
+                    "id": "email",
+                    "value": [
+                        {{
+                            "id": "[MOBILE-WORK-PERSONAL-MAIN-OTHER]",
+                            "value": "STRING"
+                        }}
+                    ]
+                }},
+                {{
+                    "id": "phone",
+                    "value": [
+                        {{
+                            "id": "[MOBILE-WORK-PERSONAL-MAIN-OTHER]",
+                            "value": "STRING (ONLY NUMBERS)"
+                        }}
+                    ]
+                }},
+                {{
+                    "id": "notes",
+                    "value": "STRING"
+                }},
+                {{
+                    "id": "roleFunction",
+                    "value": "[BUYER-SELLER-SUPPLIER-PARTNER-COLLABORATOR-PROVIDER-CUSTOMER]"
+                }}
+            ]
+        }}
+        Ejemplo de salida:
+        Para el texto de entrada: "Hablé con Ana López, CEO de Innovatech. Su número es 654 321 987 y su correo es [email protected]."
+        La salida sería:
+        {{
+            "values": [
+                {{"id": "firstName", "value": "Ana"}},
+                {{"id": "lastName", "value": "López"}},
+                {{"id": "jobTitle", "value": "CEO"}},
+                {{"id": "phone", "value": [{{"id": "MOBILE", "value": "654321987"}}]}},
+                {{"id": "email", "value": [{{"id": "WORK", "value": "[email protected]"}}]}}
+            ]
+        }}
+        Solo incluye los campos detectados en el texto de entrada.
+        Procesa el siguiente texto: "{data['inputs']}"
+            """)
         tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=1000)
         return tokens