update the llm parser with json repair

app.py

@@ -3,6 +3,7 @@ from utils import (extract_wiki_id, get_wiki_details,
                    init_llm_client, split_content_into_sections,
                    get_translate_prompt)
 import json
+import json_repair
 
 # Define language options for translation
 LANGUAGES = {
@@ -62,6 +63,12 @@ def translate_content(content, article_title, artice_summary, target_lang, api_k
         temperature=0.5
     )
 
+    decoded_object = json_repair.loads(response.choices[0].message['content'])
+    if 'output_content' in decoded_object:
+        return decoded_object['output_content']
+
+    return "Error: Translation output not found in the response."
+
 def update_ui_with_sections(sections_dict):
     """
     Creates a list of components to display in the sections area

requirements.txt

@@ -1,3 +1,4 @@
 gradio==5.29.1
 wikipedia==1.4.0
 openai==1.78.1
+json-repair==0.44.1

utils/llm_prompts.py

@@ -37,7 +37,7 @@ def get_translate_prompt(article_title, artice_summary, original_content, target
         "Return a single JSON object with the following structure:\n"
         "```json\n" +
         json.dumps({
-            "translated_content": "The complete translated XML content with all tags preserved",
+            "output_content": "The complete translated XML content with all tags preserved",
         }, indent=4, ensure_ascii=False) +
         "\n```\n\n"