Update app.py

app.py

@@ -779,9 +779,21 @@ with st.sidebar:
                         st.rerun()
 
 # MAIN CONTENT AREA
-# Check for selected task ID from URL parameters
-selected_task_id = st.query_params.get('task_id', None)
+# Get selected task ID from URL parameters
+url_task_id = st.query_params.get('task_id', None)
 
+# Update session state based on URL parameter
+if url_task_id:
+    st.session_state.selected_task_id = url_task_id
+    
+# Use session state for selected task ID
+selected_task_id = st.session_state.selected_task_id
+
+# If a back button is clicked, clear selection
+if st.button("← Back to Board", key="back_btn"):
+    st.session_state.selected_task_id = None
+    st.rerun()
+    
 # If a task is selected, show task details
 if selected_task_id:
     # Get task details
@@ -790,9 +802,7 @@ if selected_task_id:
     if not task_df.empty:
         # Back to board button
         if st.button("← Back to Board", key="back_btn"):
-            # Clear the task_id parameter and rerun
-            for key in list(st.query_params.keys()):
-                del st.query_params[key]
+            st.session_state.selected_task_id = None
             st.rerun()
         
         # Get task data
@@ -958,33 +968,29 @@ else:
                         
                         # Format dates for display - shorter format
                         date_started_str = task['Date Started'].strftime('%m/%d') if 'Date Started' in task and pd.notna(task['Date Started']) else "Not set"
-                        date_to_finish_str = task['Date to Finish'].strftime('%m/%d') if 'Date to Finish' in task and pd.notna(task['Date to Finish']) else "Not set"
-                        
-                        # Create the card HTML with direct query parameter link
-                        card_html = f"""
-                        <a href="?task_id={task_id}" style="text-decoration:none; color:inherit;">
-                            <div class="task-card" data-task-id="{task_id}">
-                                <div class="card-header {status_class}">
-                                    <span class="task-id">{task_id}</span>
-                                </div>
-                                <div class="task-title">{task['Title']}</div>
-                                <div class="task-assignee">{task['Assignee']}</div>
-                                <div class="task-dates">
-                                    <span class="date-display">Start: {date_started_str}</span>
-                                    <span class="date-display">Finish: {date_to_finish_str}</span>
-                                </div>
-                            </div>
-                        </a>
-                        """
-                        
-                        # Render the card
-                        st.markdown(card_html, unsafe_allow_html=True)
-                    
-                    # Close the container
-                    st.markdown('</div>', unsafe_allow_html=True)
-        
-        # Close container
-        st.markdown('</div>', unsafe_allow_html=True)
+                        date_to_finish_str = task['Date to Finish'].strftime('%m/%d') if 'Date to Finish' in task and pd.notna(task['Date to Finish']) else "Not set"                        
+
+                        # With this direct Streamlit approach:
+                        with st.container():
+                            col1, col2 = st.columns([4, 1])
+                            with col1:
+                                st.markdown(f"""
+                                <div class="task-card" data-task-id="{task_id}">
+                                        <div class="card-header {status_class}">
+                                            <span class="task-id">{task_id}</span>
+                                        </div>
+                                        <div class="task-title">{task['Title']}</div>
+                                        <div class="task-assignee">{task['Assignee']}</div>
+                                        <div class="task-dates">
+                                            <span class="date-display">Start: {date_started_str}</span>
+                                            <span class="date-display">Finish: {date_to_finish_str}</span>
+                                        </div>
+                                    </div>
+                                    """, unsafe_allow_html=True)
+                                with col2:
+                                    if st.button("View", key=f"view_{task_id}"):
+                                        st.session_state.selected_task_id = task_id
+                                        st.rerun()
 
 # Add JavaScript for better interactivity
 st.markdown("""