pass preference prompt

app.py

@@ -54,7 +54,7 @@ def extract_wikipedia_content(wiki_url, api_key, model_id, base_url, target_lang
     )
 
 def translate_content(content, article_title, artice_summary, content_format, 
-                      target_lang, api_key, model_id, base_url, debug_mode=False):
+                      target_lang, api_key, model_id, base_url, preference_prompt=None, debug_mode=False):
 
     llm_client = init_llm_client(api_key, base_url=base_url)
 
@@ -65,7 +65,8 @@ def translate_content(content, article_title, artice_summary, content_format,
         artice_summary=artice_summary,
         original_content=content,
         target_lang=target_lang,
-        content_format=content_format
+        content_format=content_format,
+        preference_prompt=preference_prompt
     )
 
     # Call the LLM to get the translation - updating params to match OpenAI's requirements
@@ -104,7 +105,7 @@ 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, custom_lang, api_key, model_id, base_url, debug_mode=False):
+                      target_lang, custom_lang, api_key, model_id, base_url, preference_prompt=None, debug_mode=False):
     """
     Translates a single section of the Wikipedia article
     """
@@ -126,6 +127,7 @@ def translate_section(section_content, article_title, article_summary, content_f
         api_key=api_key,
         model_id=model_id,
         base_url=base_url,
+        preference_prompt=preference_prompt,
         debug_mode=debug_mode
     )
     
@@ -281,6 +283,15 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                     value=False,
                     info="Show detailed information about LLM calls"
                 )
+                
+                # Add preference prompt section
+                gr.Markdown("### Translation Preferences")
+                preference_prompt = gr.Textbox(
+                    label="Additional Translation Preferences",
+                    placeholder="Enter any specific translation preferences or instructions...",
+                    lines=5,
+                    info="Optional: Add specific preferences for how the translation should be performed"
+                )
             
             gr.Markdown("### About")
             gr.Markdown("""
@@ -370,6 +381,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                                 api_key,
                                 model_id,
                                 base_url,
+                                preference_prompt,  # Add preference prompt to inputs
                                 debug_mode
                             ],
                             outputs=[translation_output, debug_info_state]

utils/llm_prompts.py

@@ -1,9 +1,17 @@
 import json
 
-def get_translate_prompt(article_title, artice_summary, content_format, original_content, target_lang):
+def get_translate_prompt(article_title, artice_summary, content_format, original_content, target_lang, preference_prompt=None):
     """
     Function to get the translation prompt for the LLM to translate Wikipedia {content_format} content
     with high quality and fidelity to the original.
+    
+    Args:
+        article_title: Title of the Wikipedia article
+        artice_summary: Summary of the Wikipedia article
+        content_format: Format of the content (e.g., XML, Text)
+        original_content: The content to be translated
+        target_lang: The target language for translation
+        preference_prompt: Optional additional user preferences for translation
     """
     # Define the prompt template
     translate_prompt = (
@@ -31,9 +39,19 @@ def get_translate_prompt(article_title, artice_summary, content_format, original
         "4. Preserve proper nouns, scientific terminology, and citations appropriately\n"
         "5. Adapt cultural references or idioms to be understandable in the target language\n"
         "6. Use terminology consistent with the {target_lang} Wikipedia for similar topics\n"
-        "7. Maintain the same paragraph structure and information hierarchy\n\n"
-        
-        "# Output Format\n"
+        "7. Maintain the same paragraph structure and information hierarchy\n"
+    )
+    
+    # Add user preference prompt if provided
+    if preference_prompt and preference_prompt.strip():
+        translate_prompt += (
+            "\n# Additional Translation Preferences\n"
+            f"{preference_prompt}\n"
+        )
+    
+    # Add the output format section
+    translate_prompt += (
+        "\n# Output Format\n"
         "Return a single JSON object with the following structure:\n"
         "```json\n"
         "{{\n"