modified the developer function sothat it can be utilized in python script

Author: thecodacus

lazydev/develop.py

@@ -28,5 +28,6 @@ def run(args):
     directory= args.directory
     load_dotenv()
     api_key = os.environ.get('OPENAI_API_KEY')
-    developer =Developer(requirement=requirement,root_dir=directory,openai_api_key=api_key)
+    model=args.model
+    developer =Developer(requirement=requirement,root_dir=directory,openai_api_key=api_key,model=model)
     developer.develop()

lazydev/modules/developer.py

@@ -22,13 +22,13 @@
 from .utils import Utilities
 
 class Developer():
-    def __init__(self, requirement:str,root_dir:str, openai_api_key:str):
+    def __init__(self, requirement:str,root_dir:str, openai_api_key:str,model: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",
+            model=model,
             openai_api_key=self.api_key,
             temperature=0.1,
             streaming=False
@@ -48,11 +48,23 @@ def brain_storm(self,prompt:str,step_name="prompt")->str:
         Utilities.write_to_file(f"{prompt}\n\n{aiMessage.content}",f"./thoughts/{step_name}.md")
         return aiMessage.content
 
-    def clear_doubts(self):
+    def get_doubts(self):
         prompt=PromptBook.expand_requirements(self.requirement)
         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()!=""]
+        return doubt_list
+    
+    def get_clarifications(self,doubts:List[str],answers:List[str]):
+        clarifications=""
+        for i in range(len(doubts)):
+            clarifications=f"{clarifications}\n\n{i+1}. {doubts[i]}\n Ans: {answers[i]}"
+        self.clarifications=clarifications
+        return clarifications
+    
+
+    def clear_doubts(self):
+        doubt_list=self.get_doubts()
         print("""
 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. 
@@ -73,10 +85,11 @@ def clear_doubts(self):
 Cheers! 👨‍💻
         """)
         return doubt_list,answer_list
-
+    
     def plan_project(self):
         prompt=PromptBook.plan_project(self.requirement,self.clarifications)
         plannings:str=self.brain_storm(prompt,'plan-project')
+        self.plannings=plannings
         return plannings
 
     def generate_folder_structure(self):
@@ -90,7 +103,7 @@ def generate_folder_structure(self):
             try:
 
                 folder_tree_str:str=self.brain_storm(prompt,"generate-filders")
-                folder_tree:dict=json.loads(folder_tree_str)
+                folder_tree:dict=json.loads(folder_tree_str.strip().strip("`"))
                 break
             except:
                 print("Opps messed up the json format, let me try again")
@@ -100,6 +113,7 @@ def generate_folder_structure(self):
             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)
+        return self.root_folder_name, self.file_paths
 
 
     def prioratize_files(self):
@@ -116,14 +130,16 @@ def prioratize_files(self):
             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")
+        return self.file_paths
 
     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
+            file_path_to_write=file_path,
+            file_paths=self.file_paths
             )
         code=self.brain_storm(prompt,f'code-{file_path.split("/")[-1]}') 
         Utilities.write_to_file(code,file_path=file_path)
@@ -135,20 +151,21 @@ def write_file_content(self,file_path):
     def develop(self):
         # clearing all doubts
         doubts,answers=self.clear_doubts()
-        self.clarifications:str=""
-        for i in range(len(doubts)):
-            self.clarifications=f"{self.clarifications}\n\n{i+1}. {doubts[i]}\n Ans: {answers[i]}"
-
+        self.clarifications=self.get_clarifications(doubts=doubts,answers=answers)
         # planning the project
         print("Planning...")
-        self.plannings=self.plan_project()
+        self.plan_project()
+        print(self.plannings)
+        print("\n\n")
         # 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]
+            if(file_name.split(".")[-1] in ["png","jpg","jpeg","bimp"]):
+                continue
             print(f"\nWriting Code for :{file_name}")
             self.write_file_content(file_path)
 

lazydev/modules/prompts.py

@@ -127,8 +127,11 @@ def write_file(question,clarifications:str,plan:str,files_written:List[List[str]
 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
+* if the language support, add comments at steps, which expains what you are about to do, dont add comment if comment is not supported by the file type example json file
 * keep in mind there wont be any additional files other then the full files list given above, only use files that are mentioned in that list
 Begin!
+
+File:{file_path_to_write}
+Content:
 """
 

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="lazydev",
-    version='0.0.3',
+    version='0.0.6',
     packages=find_packages(),
     install_requires=[
         "langchain>=0.0.188",