Update app2.py

app2.py

@@ -7,6 +7,10 @@ import subprocess
 import git
 import logging
 
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 # API Key Input
 if "GOOGLE_API_KEY" not in st.session_state:
     st.session_state.GOOGLE_API_KEY = ""
@@ -17,6 +21,7 @@ st.session_state.GOOGLE_API_KEY = st.text_input("API Key:", value=st.session_sta
 # Initialize Google Search API Wrapper (only if API key is provided)
 if st.session_state.GOOGLE_API_KEY:
     search = GoogleSearchAPIWrapper(google_api_key=st.session_state.GOOGLE_API_KEY)
+    
     # Agents
     agents = {
         "WEB_DEV": {
@@ -108,7 +113,6 @@ 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...")
-            # 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")
@@ -141,44 +145,43 @@ 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) 
+            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:
             st.warning("Please enter a project name")
 
     # 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
-        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)
-            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:
-            st.info("No files in the project. Use the chat interface to generate code.")
-else:
-    st.info("No projects created yet. Create a project to start coding.")
+    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
+            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)
+                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
+                            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:
+                st.info("No files in the project. Use the chat interface to generate code.")
     else:
         st.info("No projects created yet. Create a project to start coding.")
 
     # Terminal Interface
     st.subheader("Terminal (Workspace Context)")
     if st.session_state.workspace_projects:
-        selected_project = st.selectbox("Select project for terminal", list(st.session_state.workspace_projects.keys()))
+        selected_project = st.selectbox("Select project for terminal", 
+                                      list(st.session_state.workspace_projects.keys()),
+                                      key="terminal_project_select")
         terminal_input = st.text_input("Enter a command within the workspace:")
         if st.button("Run Command"):
             terminal_output = terminal_interface(terminal_input, selected_project)
@@ -239,7 +242,9 @@ else:
 
     # Collaborative Agent Example
     st.subheader("Collaborative Agent Example")
-    collab_agents = st.multiselect("Select AI agents for collaboration", list(agents.keys()), key="collab_agent_select")
+    collab_agents = st.multiselect("Select AI agents for collaboration", 
+                                  list(agents.keys()), 
+                                  key="collab_agent_select")
     collab_project = st.text_input("Enter project name for collaboration:")
     collab_task = st.text_input("Enter collaborative task:")
     if st.button("Start Collaborative Task"):
@@ -249,4 +254,4 @@ else:
             st.warning("Please select agents, enter a project name, and a task.")
 
 else:
-    st.warning("Please enter your Google Search API Key to continue.")
+    st.warning("Please enter your Google Search API Key to continue.")