From 449bf185f97ad909288e7403df2999e70e13b899 Mon Sep 17 00:00:00 2001
From: sublimegame <85697406+sublimegame@users.noreply.github.com>
Date: Wed, 16 Apr 2025 15:36:28 +0700
Subject: [PATCH] Add language parameter for multi-language support

---
 main.py | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/main.py b/main.py
index d140797..7f4f631 100644
--- a/main.py
+++ b/main.py
@@ -8,32 +8,34 @@ dotenv.load_dotenv()
 
 # Default file patterns
 DEFAULT_INCLUDE_PATTERNS = {
-    "*.py", "*.js", "*.jsx", "*.ts", "*.tsx", "*.go", "*.java", "*.pyi", "*.pyx", 
-    "*.c", "*.cc", "*.cpp", "*.h", "*.md", "*.rst", "Dockerfile", 
+    "*.py", "*.js", "*.jsx", "*.ts", "*.tsx", "*.go", "*.java", "*.pyi", "*.pyx",
+    "*.c", "*.cc", "*.cpp", "*.h", "*.md", "*.rst", "Dockerfile",
     "Makefile", "*.yaml", "*.yml",
 }
 
 DEFAULT_EXCLUDE_PATTERNS = {
-    "*test*", "tests/*", "docs/*", "examples/*", "v1/*", 
-    "dist/*", "build/*", "experimental/*", "deprecated/*", 
+    "*test*", "tests/*", "docs/*", "examples/*", "v1/*",
+    "dist/*", "build/*", "experimental/*", "deprecated/*",
     "legacy/*", ".git/*", ".github/*", ".next/*", ".vscode/*", "obj/*", "bin/*", "node_modules/*", "*.log"
 }
 
 # --- Main Function ---
 def main():
     parser = argparse.ArgumentParser(description="Generate a tutorial for a GitHub codebase or local directory.")
-    
+
     # Create mutually exclusive group for source
     source_group = parser.add_mutually_exclusive_group(required=True)
     source_group.add_argument("--repo", help="URL of the public GitHub repository.")
     source_group.add_argument("--dir", help="Path to local directory.")
-    
+
     parser.add_argument("-n", "--name", help="Project name (optional, derived from repo/directory if omitted).")
     parser.add_argument("-t", "--token", help="GitHub personal access token (optional, reads from GITHUB_TOKEN env var if not provided).")
     parser.add_argument("-o", "--output", default="output", help="Base directory for output (default: ./output).")
     parser.add_argument("-i", "--include", nargs="+", help="Include file patterns (e.g. '*.py' '*.js'). Defaults to common code files if not specified.")
     parser.add_argument("-e", "--exclude", nargs="+", help="Exclude file patterns (e.g. 'tests/*' 'docs/*'). Defaults to test/build directories if not specified.")
     parser.add_argument("-s", "--max-size", type=int, default=100000, help="Maximum file size in bytes (default: 100000, about 100KB).")
+    # Add language parameter for multi-language support
+    parser.add_argument("--language", default="english", help="Language for the generated tutorial (default: english)")
 
     args = parser.parse_args()
 
@@ -57,6 +59,9 @@ def main():
         "exclude_patterns": set(args.exclude) if args.exclude else DEFAULT_EXCLUDE_PATTERNS,
         "max_file_size": args.max_size,
 
+        # Add language for multi-language support
+        "language": args.language,
+
         # Outputs will be populated by the nodes
         "files": [],
         "abstractions": [],
@@ -66,13 +71,14 @@ def main():
         "final_output_dir": None
     }
 
-    print(f"Starting tutorial generation for: {args.repo or args.dir}")
+    # Display starting message with repository/directory and language
+    print(f"Starting tutorial generation for: {args.repo or args.dir} in {args.language.capitalize()} language")
 
     # Create the flow instance
     tutorial_flow = create_tutorial_flow()
 
     # Run the flow
     tutorial_flow.run(shared)
-    
+
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()