download translated contents

app.py

@@ -196,11 +196,14 @@ def generate_download_original(download_format, article_title, article_summary,
         f.write(content)
     return temp_path
 
-def generate_download_translated(download_format, article_title, article_summary, translations):
+def generate_download_translated(download_format, article_title, article_summary, sections, *translations_values):
     """
-    Generate a downloadable translated content file in the specified format.
+    Generate downloadable translated content file from existing translations.
     """
-    # Prepare content and filename
+    # Build translations dict from provided UI values
+    section_titles = list(sections.keys())
+    translations = {section_titles[i]: translations_values[i] for i in range(min(len(section_titles), len(translations_values)))}
+    # Build downloadable content
     if download_format == "Wikipedia XML":
         parts = [f"<article title=\"{article_title}\">"]
         for title, text in translations.items():
@@ -219,13 +222,13 @@ def generate_download_translated(download_format, article_title, article_summary
         obj = {"title": article_title, "summary": article_summary, "sections": translations}
         content = json.dumps(obj, ensure_ascii=False, indent=2)
         filename = f"{article_title or 'article'}_translated.json"
-    else:  # Plain Text
+    else:
         parts = [article_title or 'Article', article_summary or '']
         for title, text in translations.items():
             parts.append(f"## {title}\n{text}")
         content = "\n\n".join(parts)
         filename = f"{article_title or 'article'}_translated.txt"
-    # Write to a temp file and return its path
+    # Write to temp file
     temp_path = os.path.join(tempfile.gettempdir(), filename)
     with open(temp_path, 'w', encoding='utf-8') as f:
         f.write(content)
@@ -532,9 +535,11 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         inputs=[download_format, article_title, aticle_summary, article_xml, sections_state],
         outputs=[download_original_file]
     )
+    # Prepare existing translation outputs for download
+    translation_outputs = [section_components[i+2] for i in range(0, len(section_components), 4)]
     download_translated_btn.click(
         fn=generate_download_translated,
-        inputs=[download_format, article_title, aticle_summary, sections_state],
+        inputs=[download_format, article_title, aticle_summary, sections_state] + translation_outputs,
         outputs=[download_translated_file]
     )