Update app2.py

app2.py

@@ -149,26 +149,29 @@ if st.session_state.GOOGLE_API_KEY:
             st.warning("Please enter a project name")
 
     # Code Editor
-    st.subheader("Code Editor")
-    if st.session_state.workspace_projects:
-        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
-                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.")
+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.")
     else:
         st.info("No projects created yet. Create a project to start coding.")