Update app2.py

app2.py

@@ -1,9 +1,11 @@
+import os
 import streamlit as st
-from langchain.utilities import GoogleSearchAPIWrapper  # Corrected import
+from langchain.utilities import GoogleSearchAPIWrapper
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 import subprocess
 import git
+import logging
 
 # API Key Input
 if "GOOGLE_API_KEY" not in st.session_state:
@@ -73,10 +75,14 @@ if st.session_state.GOOGLE_API_KEY:
             return f"An unexpected error occurred while running the command."
 
     def get_agent_response(message: str, system_prompt: str):
-        llm = GoogleSearchAPIWrapper(google_api_key=st.session_state.GOOGLE_API_KEY)  # Access API key from session state
+        # Access API key from session state
+        llm = GoogleSearchAPIWrapper(google_api_key=st.session_state.GOOGLE_API_KEY)  
         memory = ConversationBufferMemory()
         conversation = ConversationChain(llm=llm, memory=memory)
-        response = conversation.run(system_prompt + "\n" + message)
+        
+        # Combine system prompt and user message
+        full_prompt = f"{system_prompt}\n{message}" 
+        response = conversation.run(full_prompt)
         return response
 
     def display_agent_info(agent_name: str):
@@ -102,7 +108,9 @@ if st.session_state.GOOGLE_API_KEY:
         st.info("Starting autonomous build process...")
         for agent in selected_agents:
             st.write(f"Agent {agent} is working on the project...")
-            code = get_agent_response(f"Generate code for a simple web application in project {project_name}", agents[agent]['system_prompt'])
+            # More specific prompt for code generation
+            prompt = f"Generate Python code for a simple web application using Flask framework in project {project_name}. Include instructions for running the application."  
+            code = get_agent_response(prompt, agents[agent]['system_prompt'])
             add_code_to_workspace(project_name, code, f"{agent.lower()}_app.py")
             st.write(f"Agent {agent} has completed its task.")
         st.success("Autonomous build process completed!")
@@ -122,6 +130,7 @@ if st.session_state.GOOGLE_API_KEY:
     def combine_and_process_responses(responses: dict[str, str], task: str) -> str:
         combined = "\n\n".join([f"{agent}: {response}" for agent, response in responses.items()])
         return f"Combined response for task '{task}':\n\n{combined}"
+
     # Streamlit UI
     st.title("DevToolKit: AI-Powered Development Environment")
 
@@ -132,6 +141,8 @@ if st.session_state.GOOGLE_API_KEY:
         if project_name and project_name not in st.session_state.workspace_projects:
             st.session_state.workspace_projects[project_name] = {'files': []}
             st.success(f"Created project: {project_name}")
+            # Create the project directory
+            os.makedirs(project_name, exist_ok=True) 
         elif project_name in st.session_state.workspace_projects:
             st.warning(f"Project {project_name} already exists")
         else:
@@ -140,17 +151,18 @@ if st.session_state.GOOGLE_API_KEY:
     # Code Editor
     st.subheader("Code Editor")
     if st.session_state.workspace_projects:
-        selected_project = st.selectbox("Select project", list(st.session_state.workspace_projects.keys()))
-        if selected_project:
-            files = [file['file_name'] for file in st.session_state.workspace_projects[selected_project]['files']]
-            selected_file = st.selectbox("Select file to edit", files) if files else None
+        selected_project =            st.selectbox("Select file to edit", files) if files else None
             if selected_file:
                 file_content = next((file['code'] for file in st.session_state.workspace_projects[selected_project]['files'] if file['file_name'] == selected_file), "")
-               edited_code = st.text_area("Edit code", value=file_content, height=300)  # Using st.text_area for now
+                edited_code = st.text_area("Edit code", value=file_content, height=300)  # Using st.text_area for now
                 if st.button("Save Changes"):
                     for file in st.session_state.workspace_projects[selected_project]['files']:
                         if file['file_name'] == selected_file:
                             file['code'] = edited_code
+                            # Save the changes to the actual file in the project directory
+                            file_path = os.path.join(selected_project, selected_file) 
+                            with open(file_path, "w") as f:
+                                f.write(edited_code)
                             st.success("Changes saved successfully!")
                             break
             else: