set custom language

app.py

@@ -18,13 +18,18 @@ LANGUAGES = {
     "Japanese": "ja",
     "Chinese": "zh",
     "Hindi": "hi",
-    "Korean": "ko"
+    "Korean": "ko",
+    "Custom": "custom"  # Add custom option
 }
 
-def extract_wikipedia_content(wiki_url, api_key, model_id, base_url, target_lang, content_format):
+def extract_wikipedia_content(wiki_url, api_key, model_id, base_url, target_lang, custom_lang, content_format):
     """
     Function to extract content from Wikipedia URL (placeholder for now)
     """
+    # Use custom language if selected
+    if target_lang == "custom" and custom_lang:
+        target_lang = custom_lang
+        
     wiki_id = extract_wiki_id(wiki_url)
     if not wiki_id:
         return "Invalid Wikipedia URL. Please check the URL and try again.", None, None, None, None, {}
@@ -50,6 +55,8 @@ def translate_content(content, article_title, artice_summary, content_format,
 
     llm_client = init_llm_client(api_key, base_url=base_url)
 
+    # Use the target_lang as is - it should already be properly resolved
+    # by the calling function (either a language code or custom value)
     translation_prompt = get_translate_prompt(
         article_title=article_title,
         artice_summary=artice_summary,
@@ -74,19 +81,25 @@ def translate_content(content, article_title, artice_summary, content_format,
 
     return "Error: Translation output not found in the response."
 
-def translate_section(section_content, article_title, article_summary, content_format, target_lang, api_key, model_id, base_url):
+def translate_section(section_content, article_title, article_summary, content_format, target_lang, custom_lang, api_key, model_id, base_url):
     """
     Translates a single section of the Wikipedia article
     """
     if not section_content or not api_key:
         return "Please provide content and API key for translation."
     
+    # Use custom language if selected
+    if target_lang == "custom" and custom_lang:
+        actual_lang = custom_lang
+    else:
+        actual_lang = target_lang
+    
     return translate_content(
         content=section_content,
         article_title=article_title,
         artice_summary=article_summary,
         content_format=content_format,
-        target_lang=target_lang,
+        target_lang=actual_lang,
         api_key=api_key,
         model_id=model_id,
         base_url=base_url
@@ -160,6 +173,12 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
             gr.update(visible=not new_expanded)  # Control visibility of the show button
         )
     
+    # Function to show/hide custom language input based on selection
+    def toggle_custom_language(target_lang):
+        if target_lang == "Custom":
+            return gr.update(visible=True)
+        return gr.update(visible=False)
+    
     with gr.Row() as main_layout:
         # Sidebar for configuration
         with gr.Column(scale=1, visible=True) as sidebar:
@@ -193,6 +212,20 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                     label="Target Language",
                 )
                 
+                custom_language = gr.Textbox(
+                    label="Custom Language",
+                    placeholder="Enter language name (e.g., Swedish, Dutch, etc.)",
+                    visible=False,
+                    info="Specify your desired language if not in the list above"
+                )
+                
+                # Connect the dropdown to show/hide custom language input
+                target_language.change(
+                    fn=toggle_custom_language,
+                    inputs=[target_language],
+                    outputs=[custom_language]
+                )
+                
                 content_format = gr.Radio(
                     choices=["Text", "XML"],
                     value="XML",
@@ -273,6 +306,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                                 aticle_summary,
                                 content_format,
                                 target_language,
+                                custom_language,
                                 api_key,
                                 model_id,
                                 base_url
@@ -283,7 +317,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     # Connect the extract button to the function
     extract_button.click(
         fn=extract_wikipedia_content,
-        inputs=[wiki_url, api_key, model_id, base_url, target_language, content_format],
+        inputs=[wiki_url, api_key, model_id, base_url, target_language, custom_language, content_format],
         outputs=[
             output,
             article_pageid,