code generation added

Author: thecodacus

.gitignore

@@ -1,4 +1,5 @@
 .env
 __pycache__
 code
-.venv
+.venv
+thoughts

lazydev/modules/developer.py

@@ -1,5 +1,7 @@
 
+import sys
 from typing import List
+import os
 from .prompts import PromptBook
 from langchain.llms import OpenAI
 from langchain.chat_models import ChatOpenAI
@@ -24,28 +26,35 @@ def __init__(self, requirement:str,root_dir:str, openai_api_key:str):
         self.requirement=requirement
         self.root_dir=root_dir
         self.api_key=openai_api_key
+        self.files_written=[]
         self.brain = ChatOpenAI(
             model="gpt-3.5-turbo",
             openai_api_key=self.api_key,
-            temperature=0,
+            temperature=0.1,
             streaming=False
             )
-        
-    def brain_storm(self,prompt:str)->str:
+    
+
+    def brain_storm(self,prompt:str,step_name="prompt")->str:
         messages = [
             SystemMessage(content="you are a senior software developer"),
             HumanMessage(content=prompt)
         ]
         aiMessage:AIMessage= self.brain(messages=messages)
+        if not os.path.exists("./thoughts"):
+            os.makedirs("./thoughts")
+        if os.path.exists(f"./thoughts/{step_name}.md"):
+            os.remove(f"./thoughts/{step_name}.md")
+        Utilities.write_to_file(f"{prompt}\n\n{aiMessage.content}",f"./thoughts/{step_name}.md")
         return aiMessage.content
 
     def clear_doubts(self):
         prompt=PromptBook.expand_requirements(self.requirement)
-        doubts=self.brain_storm(prompt)
+        doubts=self.brain_storm(prompt,'clear-doubts')
         doubt_list:List[str]=doubts.split("\n")
         doubt_list=[doubt.strip() for doubt in doubt_list if doubt.strip()!=""]
         print("""
-Hey there! 😄 It's Lazy Dev, your friendly neighborhood programmer, here to make your API service dreams come true! 🎉 
+Hey there! 😄 It's Lazy Dev, your friendly neighborhood programmer, here to make your awesome project's dreams come true! 🎉 
 But before I dive into coding magic, I have a few fun and important questions for you. 
 So, grab a cup of coffee ☕️, sit back, and let's clarify some details, shall we? Here we go! 🚀
         """)
@@ -67,7 +76,7 @@ def clear_doubts(self):
 
     def plan_project(self):
         prompt=PromptBook.plan_project(self.requirement,self.clarifications)
-        plannings:str=self.brain_storm(prompt)
+        plannings:str=self.brain_storm(prompt,'plan-project')
         return plannings
 
     def generate_folder_structure(self):
@@ -76,9 +85,52 @@ def generate_folder_structure(self):
             plan=self.plannings,
             clarifications=self.clarifications
             )
-        folder_tree_str:str=self.brain_storm(prompt)
-        folder_tree:dict=json.loads(folder_tree_str)
-        Utilities.generate_files_and_folders(structure=folder_tree,root_dir=self.root_dir)
+        retry_count=3
+        while retry_count>0:
+            try:
+
+                folder_tree_str:str=self.brain_storm(prompt,"generate-filders")
+                folder_tree:dict=json.loads(folder_tree_str)
+                break
+            except:
+                print("Opps messed up the json format, let me try again")
+                retry_count=retry_count-1
+        
+        if retry_count==0:
+            print("Sorry I was not able to create the folder structure in json correct format, check my instructions and try to refine it sothat i can understan the task better")
+            sys.exit()       
+        self.root_folder_name, self.file_paths = Utilities.generate_files_and_folders(structure=folder_tree,root_dir=self.root_dir)
+
+
+    def prioratize_files(self):
+        prompt=PromptBook.prioritise_file_list(self.file_paths)
+        retry_count=3
+        while retry_count>0:
+            try:
+                file_paths_str=self.brain_storm(prompt,'prioratize_files') 
+                break
+            except:
+                print("Opps messed up the json format, let me try again")
+                retry_count=retry_count-1
+        if retry_count==0:
+            print("Sorry I was not able to create the file list in correct format, check my instructions and try to refine it sothat i can understan the task better")
+            sys.exit()       
+        self.file_paths=file_paths_str.split("\n")
+    
+    def write_file_content(self,file_path):
+        prompt=PromptBook.write_file(
+            question=self.requirement,
+            clarifications=self.clarifications,
+            plan=self.plannings,
+            files_written=self.files_written,
+            file_path_to_write=file_path
+            )
+        code=self.brain_storm(prompt,f'code-{file_path.split("/")[-1]}') 
+        Utilities.write_to_file(code,file_path=file_path)
+        self.files_written.append((
+            file_path,
+            code
+        ))
 
     def develop(self):
         # clearing all doubts
@@ -93,6 +145,13 @@ def develop(self):
         # creating files and folders for the project
         print("Creating files...")
         self.generate_folder_structure()
+        self.prioratize_files()
+        self.files_written=[]
+        for file_path in self.file_paths:
+            file_name=file_path.split("/")[-1]
+            print(f"\nWriting Code for :{file_name}")
+            self.write_file_content(file_path)
+
 
 
 

lazydev/modules/prompts.py

@@ -36,12 +36,11 @@ def plan_project(question:str, clarification:str)->str:
 your current task is to design the folder and and files structure for this project
 you should follow the below format to answer this question 
 
-Plan : <your plan here> //tell us a very detailed, verbose plan or thought on how you want to proceed before writing the files  for example which framework you are going to use
+Plan : <your plan here> //tell us a very detailed, verbose plan or thought on how you want to proceed before writing the files  for example which framework you are going to use,dont forget to mention framewark related configuration files. Discuss project architecture so junior developer can understand very easily
 
 As your response will go to an automated parser, things to keep in mind all the time:
 * follow the exact format provided above without fail
-* the folder and files structure should be complete, means means no additional files will be required to create to run the program, for example if its nodejs project add package.json if its a python project add requirements.txt etc..
-* do not write the contents of the files that will be done by the next devepoler
+* be detailed so junior developer can understand very easily
 Begin!
 """
     @staticmethod
@@ -95,4 +94,35 @@ def prioritise_file_list(filelist:List[str])->List[str]:
 * do not write any explanation
 Begin!
         """
+    @staticmethod
+    def write_file(question,clarifications:str,plan:str,files_written:List[List[str]], file_path_to_write:str)->str:
+        file_with_conent="\n\n".join([f"File:{file_path}\nContent:\n{content}" for file_path,content in files_written])
+        return f"""
+you are a senior programmer below is what your client have asked you to do:
+---
+{question}
+---
+here are some clarrification on the requirements
+---
+{clarifications}
+---
+below is the what you have already planed what to do:
+---
+{plan}
+---
+
+you are now writing the code below are the files that already written with content as follows:
+---
+{file_with_conent}
+---
+
+now you are about to write content the following file:
+{file_path_to_write}
+
+As your response will go to an automated parser, things to keep in mind all the time:
+* follow the exact format provided above without fail
+* only write the file content, no expiation, no pretext.
+* always add comments at the begening, which expains what you are about to do
+Begin!
+"""
 

lazydev/modules/utils.py

@@ -9,26 +9,38 @@ def touch(path):
 
     @staticmethod
     def generate_files_recursively(files:List[Dict],cur_dir:str):
+        file_paths=[]
         for file in files:
             name=file['name']
             type=file['type']
             item_path=os.path.join(cur_dir,name)
             if type=="file":
                 Utilities.touch(item_path)
+                file_paths.append(item_path)
                 print(f"Created File: {item_path}")
             else:
                 if not os.path.exists(item_path):
                     os.makedirs(item_path)
                     print(f"Created Directory: {item_path}")
                 subfiles:List[Dict]=file['files']
-                Utilities.generate_files_recursively(subfiles,item_path)
+                sub_file_paths=Utilities.generate_files_recursively(subfiles,item_path)
+                file_paths=file_paths+sub_file_paths
+        return file_paths
 
     @staticmethod
     def generate_files_and_folders(structure:dict,root_dir:str):
         root_dir_name=structure['root_dir_name']
         cur_dir=os.path.join(root_dir,root_dir_name)
+        root_dir_path=cur_dir
         if not os.path.exists(cur_dir):
             os.makedirs(cur_dir)
             print(f"Created Directory: {cur_dir}")
         files:List[Dict]=structure['files']
-        Utilities.generate_files_recursively(files=files,cur_dir=cur_dir)
+        file_paths=Utilities.generate_files_recursively(files=files,cur_dir=cur_dir)
+        return root_dir_path, file_paths
+
+    def write_to_file(content:str,file_path:str):
+        # Open the file in write mode
+        with open(file_path, "w") as file:
+            # Write the content to the file
+            file.write(content)