import google.generativeai as genaiimport osimport jsonimport refrom pathlib import Pathfrom typing import List, Dict

class TinyDev:   """   TinyDev: A lightweight AI code generator inspired by smol-dev   Uses Gemini API to generate complete applications from simple prompts   Follows the proven three-phase workflow: Plan → Files → Code   """     def __init__(self, api_key: str, model: str = "gemini-1.5-flash"):       genai.configure(api_key=api_key)       self.model = genai.GenerativeModel(model)       self.generation_config = {           'temperature': 0.1,           'top_p': 0.8,           'max_output_tokens': 8192,       }     def plan(self, prompt: str) -> str:       """       Phase 1: Generate project plan and shared dependencies       Creates the foundation for consistent code generation       """       planning_prompt = f"""As an AI developer, you’re building a tool that automatically generates code tailored to the user’s needs.the program you are writing is based on the following description:{prompt}the files we write will be generated by a python script. the goal is for us to all work together to write a program that will write the code for the user.since we are working together, we need to understand what our shared dependencies are. this includes:- import statements we all need to use- variable names that are shared between files- functions that are called from one file to another- any other shared statethis is the most critical part of the process, if we don't get this right, the generated code will not work properly.please output a markdown file called shared_dependencies.md that lists all of the shared dependencies.the dependencies should be organized as:1. shared variables (globals, constants)2. shared functions (function signatures)3. shared classes (class names and key methods)4. shared imports (modules to import)5. shared DOM element ids (if web project)6. shared file paths/namesbe EXHAUSTIVE in your analysis. every file must be able to import or reference these shared items."""       response = self.model.generate_content(           planning_prompt,           generation_config=self.generation_config       )       return response.text   def specify_file_paths(self, prompt: str, shared_deps: str) -> List[str]:       """       Phase 2: Determine what files need to be created       """       files_prompt = f"""As an AI developer, you’re building a tool that automatically generates code tailored to the user’s needs.the program:{prompt}the shared dependencies:{shared_deps}Based on the program description and shared dependencies, return a JSON array of the filenames that should be written.Only return the JSON array, nothing else. The JSON should be an array of strings representing file paths.For example, for a simple web app you might return:["index.html", "styles.css", "script.js"]For a Python project you might return:["main.py", "utils.py", "config.py", "requirements.txt"]JSON array:"""       response = self.model.generate_content(           files_prompt,           generation_config=self.generation_config       )             try:           json_match = re.search(r'\[.*?\]', response.text, re.DOTALL)           if json_match:               files = json.loads(json_match.group())               return [f for f in files if isinstance(f, str)]           else:               lines = [line.strip() for line in response.text.split('\n') if line.strip()]               files = []               for line in lines:                   if '.' in line and not line.startswith('#'):                       file = re.sub(r'[^\w\-_./]', '', line)                       if file:                           files.append(file)               return files[:10]        except Exception as e:           print(f"Error parsing files: {e}")           return ["main.py", "README.md"]   def generate_code_sync(self, prompt: str, shared_deps: str, filename: str) -> str:       """       Phase 3: Generate code for individual files       """       code_prompt = f"""As an AI developer, you’re building a tool that automatically generates code tailored to the user’s needs..the program:{prompt}the shared dependencies:{shared_deps}Please write the file {filename}.Remember that your job is to write the code for {filename} ONLY. Do not write any other files.the code should be fully functional. meaning:- all imports should be correct- all variable references should be correct - all function calls should be correct- the code should be syntactically correct- the code should be logically correctMake sure to implement every part of the functionality described in the program description.DO NOT include ``` code fences in your response. Return only the raw code.Here is the code for {filename}:"""       response = self.model.generate_content(           code_prompt,           generation_config=self.generation_config       )             code = response.text       code = re.sub(r'^```[\w]*\n', '', code, flags=re.MULTILINE)       code = re.sub(r'\n```$', '', code, flags=re.MULTILINE)             return code.strip()   def create_app(self, prompt: str, output_dir: str = "/content/generated_app") -> Dict:       """       Main workflow: Transform a simple prompt into a complete application       """       print(f" TinyDev workflow starting...")       print(f" Prompt: {prompt}")             print("\n Step 1: Planning shared dependencies...")       shared_deps = self.plan(prompt)       print(" Dependencies planned")             print("\n Step 2: Determining file structure...")       file_paths = self.specify_file_paths(prompt, shared_deps)       print(f" Files to generate: {file_paths}")             Path(output_dir).mkdir(parents=True, exist_ok=True)             print(f"\n Step 3: Generating {len(file_paths)} files...")       results = {           'prompt': prompt,           'shared_deps': shared_deps,           'files': {},           'output_dir': output_dir       }             with open(Path(output_dir) / "shared_dependencies.md", 'w') as f:           f.write(shared_deps)             for filename in file_paths:           print(f"   Generating {filename}...")           try:               code = self.generate_code_sync(prompt, shared_deps, filename)                             file_path = Path(output_dir) / filename               file_path.parent.mkdir(parents=True, exist_ok=True)                             with open(file_path, 'w', encoding='utf-8') as f:                   f.write(code)                             results['files'][filename] = code               print(f"   {filename} created ({len(code)} chars)")                         except Exception as e:               print(f"   Error generating {filename}: {e}")               results['files'][filename] = f"# Error: {e}"             readme = f"""# Generated by TinyDev (Gemini-Powered)## Original Prompt{prompt}## Generated Files{chr(10).join(f'- {f}' for f in file_paths)}## About TinyDevTinyDev is inspired by smol-ai/developer but uses free Gemini API.It follows the proven three-phase workflow: Plan → Files → Code## UsageCheck individual files for specific usage instructions.Generated on: {os.popen('date').read().strip()}"""             with open(Path(output_dir) / "README.md", 'w') as f:           f.write(readme)             print(f"\n Complete! Generated {len(results['files'])} files in {output_dir}")       return results

def demo_tinydev():   """Demo the TinyDev code generator"""     api_key = "Use Your API Key here"     if api_key == "YOUR_GEMINI_API_KEY_HERE":       print(" Please set your Gemini API key!")       print("Get one free at: https://makersuite.google.com/app/apikey")       return None     tiny_dev = TinyDev(api_key)     demo_prompts = [       "a simple HTML/JS/CSS tic tac toe game",       "a Python web scraper that gets the latest news from multiple sources",       "a responsive landing page for a local coffee shop with contact form",       "a Flask REST API for managing a todo list",       "a JavaScript calculator with a modern UI"   ]     print(" TinyDev - AI Code Generator")   print("=" * 50)   print("Inspired by smol-ai/developer, powered by Gemini API")   print(f"Available demo projects:")     for i, prompt in enumerate(demo_prompts, 1):       print(f"{i}. {prompt}")     demo_prompt = demo_prompts[0]    print(f"\n Running demo: {demo_prompt}")     try:       results = tiny_dev.create_app(demo_prompt)             print(f"\n Results Summary:")       print(f"   Prompt: {results['prompt']}")       print(f"   Output: {results['output_dir']}")       print(f"   Files: {len(results['files'])}")             print(f"\n Generated Files:")       for filename in results['files'].keys():           print(f"  - {filename}")             if results['files']:           preview_file = list(results['files'].keys())[0]           preview_code = results['files'][preview_file]           print(f"\n  Preview of {preview_file}:")           print("-" * 40)           print(preview_code[:400] + "..." if len(preview_code) > 400 else preview_code)           print("-" * 40)             print(f"\n This uses the same proven workflow as smol-ai/developer!")       print(f" Check {results['output_dir']} for all generated files")             return results         except Exception as e:       print(f" Demo failed: {e}")       return None

def interactive_tinydev():   """Interactive version where you can try your own prompts"""   api_key = input(" Enter your Gemini API key: ").strip()     if not api_key:       print(" API key required!")       return     tiny_dev = TinyDev(api_key)     print("\n Interactive TinyDev Mode")   print("Type your app ideas and watch them come to life!")     while True:       prompt = input("\n Describe your app (or 'quit'): ").strip()             if prompt.lower() in ['quit', 'exit', 'q']:           print(" Goodbye!")           break             if prompt:           try:               results = tiny_dev.create_app(prompt, f"/content/app_{hash(prompt) % 10000}")               print(f" Success! Check {results['output_dir']}")           except Exception as e:               print(f" Error: {e}")print(" TinyDev - AI Code Generator Ready!")print("Inspired by smol-ai/developer, powered by free Gemini API")print("\nTo run demo: demo_tinydev()")print("To try interactive mode: interactive_tinydev()")

demo_tinydev()