refactor: fix multi-project GitHub PR targeting and improve error messages

- Refactor GitHubPRAgent to accept repository parameters in constructor
- Fix PR creation to target correct repository based on selected project
- Update TocTreeHandler to support project-specific toctree files
- Improve error messages with specific project mismatch detection
- Pass project parameter from handler to ensure proper repository targeting

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

agent/handler.py

@@ -108,7 +108,7 @@ def process_file_search_handler(project: str, lang: str, k: int, history: list)
 2. Click "💾 Save Configuration" 
 3. Try "Find Files" again"""
             history.append(["File search request", response])
-            return history, "", update_status(), gr.Tabs(selected=0), gr.update(choices=[])
+            return history, "", update_status(), gr.Tabs(selected=0), gr.update(choices=[]), gr.update(visible=False)
         else:
             raise  # Re-raise non-rate-limit errors
     state.files_to_translate = (
@@ -141,14 +141,13 @@ def process_file_search_handler(project: str, lang: str, k: int, history: list)
     # Add to history
     history.append(["Please find files that need translation", response])
     cleared_input = ""
-    selected_tab = 1 if state.files_to_translate else 0
 
     # 드롭다운 choices로 쓸 파일 리스트 반환 추가
     return (
         history,
         cleared_input,
         update_status(),
-        gr.Tabs(selected=selected_tab),
+        gr.Tabs(),  # Don't change tab
         update_dropdown_choices(state.files_to_translate),
     )
 
@@ -157,7 +156,30 @@ def update_dropdown_choices(file_list):
     return gr.update(choices=file_list, value=None)
 
 
-def start_translation_process():
+def confirm_and_go_translate_handler(history):
+    """Confirm selection and go to translate tab"""
+    global state
+    
+    response = f"✅ **Selection confirmed!**\n\n🎯 **Project:** {state.selected_project}\n🌍 **Language:** {state.target_language}\n\n**➡️ Go to Tab 2 to start translation.**"
+    history.append(["Confirm selection", response])
+    return history, "", update_status(), gr.Tabs(selected=1)
+
+
+def confirm_translation_and_go_upload_handler(history):
+    """Confirm translation and go to upload PR tab"""
+    global state
+    
+    if not state.current_file_content.get("translated"):
+        response = "❌ No translation available. Please complete translation first."
+        history.append(["Upload PR request", response])
+        return history, "", update_status(), gr.Tabs()
+    
+    response = f"✅ **Translation confirmed!**\n\n📄 **File:** `{state.files_to_translate[0] if state.files_to_translate else 'Unknown'}`\n\n**➡️ Go to Tab 3 to upload PR.**"
+    history.append(["Upload PR request", response])
+    return history, "", update_status(), gr.Tabs(selected=2)
+
+
+def start_translation_process(force_retranslate=False):
     """Start the translation process for the first file"""
     if not state.files_to_translate:
         return "❌ No files available for translation.", ""
@@ -166,8 +188,8 @@ def start_translation_process():
 
     # Call translation function (simplified for demo)
     try:
-        translated = translate_docs_interactive(
-            state.target_language, [[current_file]], state.additional_instruction
+        status, translated = translate_docs_interactive(
+            state.target_language, [[current_file]], state.additional_instruction, state.selected_project, force_retranslate
         )
 
         state.current_file_content = {"translated": translated}
@@ -184,13 +206,17 @@ def start_translation_process():
         print("Compeleted translation:\n")
         print(translated)
         print("----------------------------")
-        response = (
-            f"""🔄 Translation for: `{current_file}`\n"""
-            "**📄 Original Content Link:**\n"
-            ""
-            f"{original_file_link}\n"
-            "**🌐 Translated Content:**\n"
-        )
+        
+        # Different response format for existing vs new translation
+        if isinstance(status, str) and "Existing translation loaded" in status:
+            response = f"{status}\n**📄 Original Content Link:** {original_file_link}\n\n**🌐 Translated Content:**"
+        else:
+            response = (
+                f"""🔄 Translation for: `{current_file}`\n"""
+                f"**📄 Original Content Link:** {original_file_link}\n\n"
+                f"{status}\n\n"
+                "**🌐 Translated Content:**"
+            )
         return response, translated
 
 
@@ -228,7 +254,12 @@ Currently available actions with quick controls:
     else:
         return """I understand you want to work on translations! 
 
-To get started, please use the controls above to configure your translation settings and find files that need translation.
+**Two ways to get started:**
+
+1. **🔍 Find Files first** - Use Tab 1 to discover files that need translation
+2. **🚀 Direct Translation** - Go to Tab 2 and enter a file path directly (e.g., `docs/source/en/model_doc/bert.md`)
+
+Make sure to configure your API keys in the Configuration panel above.
 """
 
 
@@ -325,7 +356,25 @@ def sync_language_displays(lang):
     return lang
 
 
-def update_persistent_config(anthropic_key, github_token, github_owner, github_repo, reference_pr_url):
+def update_project_selection(project, history):
+    """Update state when project is selected"""
+    global state
+    state.selected_project = project
+    response = f"Selection confirmed: 🎯 Project → **{project}**"
+    history.append(["Project selection", response])
+    return history, "", update_status()
+
+
+def update_language_selection(lang, history):
+    """Update state when language is selected"""
+    global state
+    state.target_language = lang
+    response = f"Selection confirmed: 🌍 Language → **{lang}**"
+    history.append(["Language selection", response])
+    return history, "", update_status(), lang
+
+
+def update_persistent_config(anthropic_key, github_token, github_owner, github_repo, reference_pr_url, history):
     """Update persistent configuration settings."""
     global state
     
@@ -353,7 +402,15 @@ def update_persistent_config(anthropic_key, github_token, github_owner, github_r
         "reference_pr_url": reference_pr_url or "",
     })
 
-    return f"✅ Configuration saved! GitHub: {github_owner}/{github_repo}"
+    # Build response message based on what was configured
+    response = "✅ Configuration saved!"
+    if github_owner and github_repo:
+        response += f" GitHub: {github_owner}/{github_repo}"
+    elif anthropic_key:
+        response += " Anthropic API key updated."
+    
+    history.append(["Configuration update", response])
+    return history, "", update_status()
 
 
 def update_github_config(token, owner, repo, reference_pr_url):
@@ -374,7 +431,7 @@ def update_prompt_preview(language, file_path, additional_instruction):
             translation_lang = language
         
         # Get sample content (first 500 characters)
-        content = get_content(file_path)
+        content = get_content(file_path, state.selected_project)
         to_translate = preprocess_content(content)
         
         # Truncate for preview
@@ -385,7 +442,10 @@ def update_prompt_preview(language, file_path, additional_instruction):
         
         return prompt
     except Exception as e:
-        return f"Error generating prompt preview: {str(e)}"
+        error_str = str(e)
+        if "Failed to retrieve content from the URL" in error_str:
+            return f"❌ **File not found:** `{file_path}`\n\n💡 **Please check:**\n1. Is this file in the **{state.selected_project}** project?\n2. Use \"🔍 Find Files to Translate\" to see available files\n3. Verify the file path is correct"
+        return f"Error generating prompt preview: {error_str}"
 
 
 def send_message(message, history):
@@ -394,21 +454,39 @@ def send_message(message, history):
 
 
 # Button handlers with tab switching
-def start_translate_handler(history, file_to_translate, additional_instruction=""):
+def start_translate_handler(history, file_to_translate, additional_instruction="", force_retranslate=False):
     # Use persistent anthropic key
     anthropic_key = state.persistent_settings["anthropic_api_key"]
     if not anthropic_key:
         response = "❌ Please set Anthropic API key in Configuration panel first."
         history.append(["Translation request", response])
-        return history, "", update_status(), gr.Tabs(selected=0)
+        return history, "", update_status(), gr.Tabs(), gr.update(), gr.update()
     
     os.environ["ANTHROPIC_API_KEY"] = anthropic_key
     
+    # Check if file path is provided
+    if not file_to_translate or not file_to_translate.strip():
+        response = "❌ Please select a file from the dropdown or enter a file path to translate."
+        history.append(["Translation request", response])
+        return history, "", update_status(), gr.Tabs(), gr.update(), gr.update()
+    
     state.additional_instruction = additional_instruction
     state.files_to_translate = [file_to_translate]
-    new_hist, cleared_input = handle_user_message("start translation", history)
-    selected_tabs = 2 if state.current_file_content["translated"] else 0
-    return new_hist, cleared_input, update_status(), gr.Tabs(selected=selected_tabs)
+    state.step = "translate"
+    
+    # Start translation directly
+    if force_retranslate:
+        history.append(["Translation request", "🔄 **Force retranslation started...**"])
+    response, translated = start_translation_process(force_retranslate)
+    history.append(["", response])
+    if translated:
+        history.append(["", translated])
+    
+    # Update button text and show confirm button after translation
+    start_btn_text = "🔄 Retranslation" if state.current_file_content["translated"] else "🚀 Start Translation"
+    confirm_btn_visible = bool(state.current_file_content["translated"])
+    
+    return history, "", update_status(), gr.Tabs(), gr.update(value=start_btn_text), gr.update(visible=confirm_btn_visible)
 
 
 def approve_handler(history, owner, repo, reference_pr_url):
@@ -416,15 +494,21 @@ def approve_handler(history, owner, repo, reference_pr_url):
     global state
     state.step = "create_github_pr"
 
-    # Use persistent settings for token, update other values
+    # Check all required GitHub configuration at once
     github_config = state.persistent_settings["github_config"]
+    missing_config = []
+    
     if not github_config.get("token"):
-        response = "❌ Please set GitHub Token in Configuration panel first."
-        history.append(["GitHub PR creation request", response])
-        return history, "", update_status()
-        
-    if not owner or not repo:
-        response = "❌ Please set GitHub Owner and Repository Name in Configuration panel first."
+        missing_config.append("GitHub Token")
+    if not owner:
+        missing_config.append("GitHub Owner")
+    if not repo:
+        missing_config.append("Repository Name")
+    
+    if missing_config:
+        config = get_project_config(state.selected_project)
+        repo_name = config.repo_url.split('/')[-1]  # Extract repo name from URL
+        response = f"❌ Please set the following in Configuration panel first: {', '.join(missing_config)}\n\n💡 **Note:** GitHub Owner/Repository should be your fork of [`{repo_name}`]({config.repo_url}) (e.g., Owner: `your-username`, Repository: `{repo_name}`)"
         history.append(["GitHub PR creation request", response])
         return history, "", update_status()
 
@@ -435,6 +519,9 @@ def approve_handler(history, owner, repo, reference_pr_url):
     # Use persistent settings
     github_config = state.persistent_settings["github_config"]
 
+    # Initialize response variable
+    response = ""
+    
     # If reference PR is not provided, use the agent to find one
     if not github_config.get("reference_pr_url"):
         response = "🤖 **Reference PR URL not found. The agent will now search for a suitable one...**"
@@ -490,6 +577,7 @@ def approve_handler(history, owner, repo, reference_pr_url):
             translated_content=translated_content,
             github_config=state.github_config,
             en_title=file_name,
+            project=state.selected_project,
         )
         response += f"\n{pr_response}"
     else:

agent/toctree_handler.py

@@ -4,9 +4,17 @@ from typing import Dict, List, Any
 import os
 
 class TocTreeHandler:
-    def __init__(self):
-        self.en_toctree_url = "https://raw.githubusercontent.com/huggingface/transformers/main/docs/source/en/_toctree.yml"
-        self.ko_toctree_url = "https://raw.githubusercontent.com/huggingface/transformers/main/docs/source/ko/_toctree.yml"
+    def __init__(self, project: str = "transformers"):
+        from translator.project_config import get_project_config
+        self.project = project
+        self.project_config = get_project_config(project)
+        
+        # Extract repository path from config
+        repo_path = self.project_config.repo_url.replace("https://github.com/", "")
+        
+        # Build project-specific URLs
+        self.en_toctree_url = f"https://raw.githubusercontent.com/{repo_path}/main/docs/source/en/_toctree.yml"
+        self.ko_toctree_url = f"https://raw.githubusercontent.com/{repo_path}/main/docs/source/ko/_toctree.yml"
         self.local_docs_path = "docs/source/ko"
     
     def fetch_toctree(self, url: str) -> Dict[str, Any]:
@@ -245,7 +253,8 @@ Korean title:"""
         translation_result: dict, 
         filepath: str, 
         pr_agent, 
-        github_config: dict
+        github_config: dict,
+        project: str = "transformers"
     ) -> dict:
         """Update toctree after successful translation PR.
         

agent/workflow.py

@@ -62,23 +62,24 @@ def report_translation_target_files(
     return status_report, [[file] for file in filepath_list]
 
 
-def translate_docs(lang: str, file_path: str, additional_instruction: str = "") -> tuple[str, str]:
+def translate_docs(lang: str, file_path: str, additional_instruction: str = "", project: str = "transformers", force_retranslate: bool = False) -> tuple[str, str]:
     """Translate documentation."""
-    # Check if translation already exists
+    # Check if translation already exists (unless force retranslate is enabled)
     translation_file_path = (
         Path(__file__).resolve().parent.parent
         / f"translation_result/{file_path}"
     )
 
-    if translation_file_path.exists():
+    if not force_retranslate and translation_file_path.exists():
         print(f"📄 Found existing translation: {translation_file_path}")
         with open(translation_file_path, "r", encoding="utf-8") as f:
             existing_content = f.read()
         if existing_content.strip():
-            return "Existing translation loaded (no tokens used). If you want to translate again, please restart the gradio app.", existing_content
+            existing_msg = f"♻️ **Existing translation loaded** (no tokens used)\n📁 **File:** `{file_path}`\n📅 **Loaded from:** `{translation_file_path}`\n💡 **To retranslate:** Check 'Force Retranslate' option."
+            return existing_msg, existing_content
 
     # step 1. Get content from file path
-    content = get_content(file_path)
+    content = get_content(file_path, project)
     to_translate = preprocess_content(content)
 
     # step 2. Prepare prompt with docs content
@@ -101,7 +102,7 @@ def translate_docs(lang: str, file_path: str, additional_instruction: str = "")
 
 
 def translate_docs_interactive(
-    translate_lang: str, selected_files: list[list[str]], additional_instruction: str = ""
+    translate_lang: str, selected_files: list[list[str]], additional_instruction: str = "", project: str = "transformers", force_retranslate: bool = False
 ) -> tuple[str, str]:
     """Interactive translation function that processes files one by one.
 
@@ -115,14 +116,22 @@ def translate_docs_interactive(
     # Start with the first file
     current_file = file_paths[0]
 
-    status = f"✅ Translation completed: `{current_file}` → `{translate_lang}`\n\n"
-    callback_result, translated_content = translate_docs(translate_lang, current_file, additional_instruction)
-    status += f"💰 Used token and cost: \n```\n{callback_result}\n```"
+    callback_result, translated_content = translate_docs(translate_lang, current_file, additional_instruction, project, force_retranslate)
+    
+    # Check if existing translation was loaded
+    if isinstance(callback_result, str) and "Existing translation loaded" in callback_result:
+        status = callback_result  # Use the existing translation message
+    else:
+        if force_retranslate:
+            status = f"🔄 **Force Retranslation completed**: `{current_file}` → `{translate_lang}`\n\n"
+        else:
+            status = f"✅ Translation completed: `{current_file}` → `{translate_lang}`\n\n"
+        status += f"💰 Used token and cost: \n```\n{callback_result}\n```"
 
     print(callback_result)
     print(status)
 
-    return translated_content
+    return status, translated_content
 
 
 def generate_github_pr(
@@ -131,6 +140,7 @@ def generate_github_pr(
     translated_content: str = None,
     github_config: dict = None,
     en_title: str = None,
+    project: str = "transformers",
 ) -> str:
     """Generate a GitHub PR for translated documentation.
 
@@ -148,7 +158,7 @@ def generate_github_pr(
         return "❌ GitHub PR Agent is not available. Please install required libraries."
 
     if not github_config:
-        return "❌ GitHub configuration not provided."
+        return "❌ GitHub configuration not provided. Please set up GitHub token, owner, and repository in Configuration panel."
 
     # Validate required configuration
     required_fields = ["token", "owner", "repo_name", "reference_pr_url"]
@@ -157,7 +167,7 @@ def generate_github_pr(
     ]
 
     if missing_fields:
-        return f"❌ Missing required configuration: {', '.join(missing_fields)}. Please provide these values."
+        return f"❌ Missing required GitHub configuration: {', '.join(missing_fields)}\n\n💡 Go to Configuration panel and set:\n" + "\n".join([f"  • {field}" for field in missing_fields])
 
     # Set token in environment for the agent.
     os.environ["GITHUB_TOKEN"] = github_config["token"]
@@ -170,29 +180,39 @@ def generate_github_pr(
                 / f"translation_result/{filepath}"
             )
             if not translation_file_path.exists():
-                return f"❌ Translation file not found: {translation_file_path}"
+                return f"❌ Translation file not found: {translation_file_path}\n\n💡 Please complete translation first in Tab 2 for file: {filepath}"
 
             with open(translation_file_path, "r", encoding="utf-8") as f:
                 translated_content = f.read()
 
         if not translated_content or not translated_content.strip():
-            return "❌ Translated content is empty."
+            return f"❌ Translated content is empty for file: {filepath}\n\n💡 Please complete translation first in Tab 2."
 
         # Execute GitHub PR Agent
+        # Get base repository from project config
+        from translator.project_config import get_project_config
+        project_config = get_project_config(project)
+        base_repo_path = project_config.repo_url.replace("https://github.com/", "")
+        base_owner, base_repo = base_repo_path.split("/")
+
         print(f"🚀 Starting GitHub PR creation...")
         print(f"   📁 File: {filepath}")
         print(f"   🌍 Language: {target_language}")
         print(f"   📊 Reference PR: {github_config['reference_pr_url']}")
-        print(f"   🏠 Repository: {github_config['owner']}/{github_config['repo_name']}")
-
-        agent = GitHubPRAgent()
+        print(f"   🏠 User Fork: {github_config['owner']}/{github_config['repo_name']}")
+        print(f"   🎯 Base Repository: {base_owner}/{base_repo}")
+
+        agent = GitHubPRAgent(
+            user_owner=github_config["owner"],
+            user_repo=github_config["repo_name"],
+            base_owner=base_owner,
+            base_repo=base_repo,
+        )
         result = agent.run_translation_pr_workflow(
             reference_pr_url=github_config["reference_pr_url"],
             target_language=target_language,
             filepath=filepath,
             translated_doc=translated_content,
-            owner=github_config["owner"],
-            repo_name=github_config["repo_name"],
             base_branch=github_config.get("base_branch", "main"),
         )
         # TEST CODE
@@ -206,9 +226,9 @@ def generate_github_pr(
         toctree_result = None
         if en_title:
             from agent.toctree_handler import TocTreeHandler
-            toctree_handler = TocTreeHandler()
+            toctree_handler = TocTreeHandler(project)
             toctree_result = toctree_handler.update_toctree_after_translation(
-                result, filepath, agent, github_config
+                result, filepath, agent, github_config, project
             )
 
         # Process result
@@ -252,13 +272,29 @@ def generate_github_pr(
 {result.get("error_details", "Unknown error")}"""
 
         else:
+            error_details = result.get("error_details", "No additional details")
             return f"""❌ **GitHub PR Creation Failed**
 
 **Error Message:**
-{result["message"]}"""
+{result["message"]}
+
+**Error Details:**
+{error_details}
+
+💡 **Common Solutions:**
+1. **Project Mismatch**: Selected project '{project}' but fork is '{github_config.get('repo_name', 'REPO')}' - ensure they match
+2. Check if your GitHub fork exists: {github_config.get('owner', 'USER')}/{github_config.get('repo_name', 'REPO')}
+3. Verify GitHub token has write access to your fork"""
 
     except Exception as e:
-        error_msg = f"❌ Unexpected error occurred during PR creation: {str(e)}"
+        error_msg = f"""❌ **Unexpected Error During PR Creation**
+
+**Error:** {str(e)}
+
+**Configuration:**
+• Project: {project}
+• File: {filepath}
+• Target: {github_config.get('owner', 'USER')}/{github_config.get('repo_name', 'REPO')} → {base_owner if 'base_owner' in locals() else 'BASE'}/{base_repo if 'base_repo' in locals() else 'REPO'}"""
         print(error_msg)
         return error_msg
 

app.py

@@ -8,12 +8,16 @@ from dotenv import load_dotenv
 
 from agent.handler import (
     approve_handler,
+    confirm_and_go_translate_handler,
+    confirm_translation_and_go_upload_handler,
     get_welcome_message,
     process_file_search_handler,
     restart_handler,
     send_message,
     start_translate_handler,
     sync_language_displays,
+    update_language_selection,
+    update_project_selection,
     update_prompt_preview,
     update_status,
     update_github_config,
@@ -188,6 +192,11 @@ with gr.Blocks(
                                 "🔍 Find Files to Translate",
                                 elem_classes="action-button",
                             )
+                            
+                            confirm_go_btn = gr.Button(
+                                "✅ Confirm Selection & Go to Translate",
+                                elem_classes="action-button",
+                            )
 
                     with gr.TabItem("2. Translate", id=1):
                         with gr.Group():
@@ -214,9 +223,13 @@ with gr.Blocks(
                                 lines=2,
                             )
                             
-                            with gr.Accordion("🔍 Preview Prompt", open=False):
+                            force_retranslate = gr.Checkbox(
+                                label="🔄 Force Retranslate (ignore existing translations)",
+                                value=False,
+                            )
+                            
+                            with gr.Accordion("🔍 Preview Translation Prompt", open=False):
                                 prompt_preview = gr.Textbox(
-                                    label="Current Translation Prompt",
                                     lines=8,
                                     interactive=False,
                                     placeholder="Select a file and language to see the prompt preview...",
@@ -226,6 +239,12 @@ with gr.Blocks(
                             start_translate_btn = gr.Button(
                                 "🚀 Start Translation", elem_classes="action-button"
                             )
+                            
+                            confirm_upload_btn = gr.Button(
+                                "✅ Confirm Translation & Upload PR",
+                                elem_classes="action-button",
+                                visible=False,
+                            )
 
                     with gr.TabItem("3. Upload PR", id=2):
                         with gr.Group():
@@ -247,12 +266,31 @@ with gr.Blocks(
         inputs=[project_dropdown, lang_dropdown, k_input, chatbot],
         outputs=[chatbot, msg_input, status_display, control_tabs, files_to_translate],
     )
+    
+    confirm_go_btn.click(
+        fn=confirm_and_go_translate_handler,
+        inputs=[chatbot],
+        outputs=[chatbot, msg_input, status_display, control_tabs],
+    )
 
-    # Sync language across tabs
+    # Auto-save selections to state and update prompt preview
+    project_dropdown.change(
+        fn=update_project_selection,
+        inputs=[project_dropdown, chatbot],
+        outputs=[chatbot, msg_input, status_display],
+    )
+    
+    # Update prompt preview when project changes
+    project_dropdown.change(
+        fn=update_prompt_preview,
+        inputs=[translate_lang_display, file_to_translate_input, additional_instruction],
+        outputs=[prompt_preview],
+    )
+    
     lang_dropdown.change(
-        fn=sync_language_displays,
-        inputs=[lang_dropdown],
-        outputs=[translate_lang_display],
+        fn=update_language_selection,
+        inputs=[lang_dropdown, chatbot],
+        outputs=[chatbot, msg_input, status_display, translate_lang_display],
     )
 
     #
@@ -265,15 +303,21 @@ with gr.Blocks(
     # Button event handlers
     start_translate_btn.click(
         fn=start_translate_handler,
-        inputs=[chatbot, file_to_translate_input, additional_instruction],
+        inputs=[chatbot, file_to_translate_input, additional_instruction, force_retranslate],
+        outputs=[chatbot, msg_input, status_display, control_tabs, start_translate_btn, confirm_upload_btn],
+    )
+    
+    confirm_upload_btn.click(
+        fn=confirm_translation_and_go_upload_handler,
+        inputs=[chatbot],
         outputs=[chatbot, msg_input, status_display, control_tabs],
     )
 
     # Configuration Save
     save_config_btn.click(
         fn=update_persistent_config,
-        inputs=[config_anthropic_key, config_github_token, config_github_owner, config_github_repo, reference_pr_url],
-        outputs=[msg_input],
+        inputs=[config_anthropic_key, config_github_token, config_github_owner, config_github_repo, reference_pr_url, chatbot],
+        outputs=[chatbot, msg_input, status_display],
     )
 
     approve_btn.click(

pr_generator/agent.py

@@ -34,9 +34,13 @@ except ImportError as e:
 class GitHubPRAgent:
     """Agent class for GitHub PR creation"""
 
-    def __init__(self):
+    def __init__(self, user_owner: str = None, user_repo: str = None, base_owner: str = None, base_repo: str = None):
         self._github_client = None
         self._llm = None
+        self.user_owner = user_owner
+        self.user_repo = user_repo
+        self.base_owner = base_owner
+        self.base_repo = base_repo
 
     @property
     def github_client(self) -> Optional[Github]:
@@ -433,8 +437,6 @@ Please return only the commit message. No other explanation is needed."""
         target_language: str,
         filepath: str,
         translated_doc: str,
-        owner: str,
-        repo_name: str,
         base_branch: str = "main",
     ) -> Dict[str, Any]:
         """Execute translation document PR creation workflow."""
@@ -458,19 +460,20 @@ Please return only the commit message. No other explanation is needed."""
             )
 
             # 3. Get main branch SHA from upstream and create branch in fork
-            upstream_repo = self.github_client.get_repo(f"huggingface/{repo_name}")
+            upstream_repo = self.github_client.get_repo(f"{self.base_owner}/{self.base_repo}")
             main_branch = upstream_repo.get_branch(base_branch)
             main_sha = main_branch.commit.sha
 
             print(f"🌿 Creating branch: {branch_name} in fork repository")
-            branch_result = self.create_branch(owner, repo_name, branch_name, main_sha)
+            branch_result = self.create_branch(self.user_owner, self.user_repo, branch_name, main_sha)
 
             # Check branch creation result
             if branch_result.startswith("ERROR"):
                 return {
                     "status": "error",
-                    "message": f"Branch creation failed: {branch_result}",
+                    "message": f"Branch creation failed: {branch_result}\n\nTarget: {self.user_owner}/{self.user_repo}\nBranch: {branch_name}\nBase SHA: {main_sha[:8]}",
                     "branch": branch_name,
+                    "error_details": branch_result,
                 }
             elif branch_result.startswith("WARNING"):
                 print(f"⚠️ {branch_result}")
@@ -489,8 +492,8 @@ Please return only the commit message. No other explanation is needed."""
 
             print(f"📄 Saving file: {target_filepath}")
             file_result = self.create_or_update_file(
-                owner,
-                repo_name,
+                self.user_owner,
+                self.user_repo,
                 target_filepath,
                 commit_message,
                 translated_doc,
@@ -500,9 +503,10 @@ Please return only the commit message. No other explanation is needed."""
             if not file_result.startswith("SUCCESS"):
                 return {
                     "status": "error",
-                    "message": "An issue occurred while saving the file.",
+                    "message": f"File save failed: {file_result}\n\n🎯 Target: {self.user_owner}/{self.user_repo} (expected: {target_language} fork of {self.base_owner}/{self.base_repo})\n🌿 Branch: {branch_name}\n📁 File: {target_filepath}",
                     "branch": branch_name,
                     "file_path": target_filepath,
+                    "error_details": file_result,
                 }
 
             print(f"{file_result}")
@@ -518,11 +522,11 @@ Please return only the commit message. No other explanation is needed."""
             )
 
             print(f"🔄 Creating PR: {pr_title}")
-            print(f"   Head: {owner}:{branch_name} → Base: huggingface:{base_branch}")
+            print(f"   Head: {self.user_owner}:{branch_name} → Base: {self.base_owner}:{base_branch}")
 
             # Create PR from fork to upstream repository
             pr_result = self.create_pull_request(
-                "huggingface", "transformers", pr_title, f"{owner}:{branch_name}", base_branch, pr_body, draft=True
+                self.base_owner, self.base_repo, pr_title, f"{self.user_owner}:{branch_name}", base_branch, pr_body, draft=True
             )
 
             if pr_result.startswith("ERROR"):
@@ -554,7 +558,8 @@ Please return only the commit message. No other explanation is needed."""
         except Exception as e:
             return {
                 "status": "error",
-                "message": f"Error occurred during workflow execution: {str(e)}",
+                "message": f"Workflow execution failed: {str(e)}\n\nConfig: {self.user_owner}/{self.user_repo} → {self.base_owner}/{self.base_repo}\nFile: {filepath if 'filepath' in locals() else 'Unknown'}",
+                "error_details": str(e),
             }
 
 

translator/content.py

@@ -6,15 +6,18 @@ from langchain.callbacks import get_openai_callback
 from langchain_anthropic import ChatAnthropic
 
 from translator.prompt_glossary import PROMPT_WITH_GLOSSARY
+from translator.project_config import get_project_config
 
 
-def get_content(filepath: str) -> str:
+def get_content(filepath: str, project: str = "transformers") -> str:
     if filepath == "":
         raise ValueError("No files selected for translation.")
 
-    url = string.Template(
-        "https://raw.githubusercontent.com/huggingface/" "transformers/main/$filepath"
-    ).safe_substitute(filepath=filepath)
+    config = get_project_config(project)
+    # Extract repo path from repo_url (e.g., "huggingface/transformers")
+    repo_path = config.repo_url.replace("https://github.com/", "")
+    
+    url = f"https://raw.githubusercontent.com/{repo_path}/main/{filepath}"
     response = requests.get(url)
     if response.status_code == 200:
         content = response.text
@@ -170,4 +173,4 @@ def llm_translate(to_translate: str) -> tuple[str, str]:
         )
         ai_message = model.invoke(to_translate)
         print("cb:", cb)
-    return cb, ai_message.content
+    return str(cb), ai_message.content