Update app.py

app.py

@@ -116,6 +116,11 @@ def workspace_page():
         st.session_state.current_project = selected_project
         st.session_state.page = "project_view"
 
+    # Display success message if a project was uploaded
+    if st.session_state.project_uploaded:
+        st.success(f"Project '{st.session_state.current_project}' uploaded successfully!")
+        st.session_state.project_uploaded = False  # Reset after showing the message
+
     # Main content area
     st.subheader("Workspace")
     st.write("You can create a new project by uploading files or folders, or by cloning a GitHub repository.")
@@ -154,10 +159,11 @@ def workspace_page():
                             st.error(f"File {uploaded_file.name} is not a valid .zip file.")
                     else:
                         st.success(f"File {uploaded_file.name} saved successfully!")
-                # Trigger a rerun to update the dropdown
-                st.rerun()
-                st.success(f"Project '{project_name}' uploaded successfully!")
 
+                # Update session state and trigger a rerun
+                st.session_state.current_project = project_name
+                st.session_state.project_uploaded = True
+                st.rerun()
 
     elif action == "Clone GitHub Repository":
         st.subheader("Clone GitHub Repository")
@@ -170,11 +176,10 @@ def workspace_page():
 
                 try:
                     Repo.clone_from(repo_url, project_folder)
-                    # Trigger a rerun to update the dropdown
+                    # Update session state and trigger a rerun
+                    st.session_state.current_project = project_name
+                    st.session_state.project_uploaded = True
                     st.rerun()
-                    st.success(f"Project '{project_name}' cloned successfully!")
-                    
-
                 except Exception as e:
                     st.error(f"Failed to clone repository: {e}")