Update app.py

app.py

@@ -454,8 +454,12 @@ if fetch_data:
 # --- Suggested Metadata Enhancements Section ---
 st.subheader("Suggested Metadata Enhancements")
 
-# Always show the checkbox
-use_ai = st.checkbox("🤖 Use AI Suggestions (Hugging Face)", value=True)
+# Create a row with checkbox for AI suggestions
+col1, col2 = st.columns([1, 7])
+with col1:
+    use_ai = st.checkbox("", value=True)
+with col2:
+    st.markdown("🤖 Use AI Suggestions (Hugging Face)")
 
 # Check if records exist
 incomplete_with_desc = metadata_df[
@@ -463,59 +467,75 @@ incomplete_with_desc = metadata_df[
     (metadata_df['subject'].isnull())
 ]
 
+# Create a consistent container for the results
+st.markdown("""
+    <div style='
+        background-color: #1e1e1e;
+        border-radius: 10px;
+        padding: 1rem;
+        margin-top: 1rem;
+    '>
+""", unsafe_allow_html=True)
+
 if not incomplete_with_desc.empty:
     if use_ai:
         suggestions = []
         records_to_process = min(10, len(incomplete_with_desc))
-
         progress = st.progress(0)
         status = st.empty()
-
+        
         for i, (idx, row) in enumerate(incomplete_with_desc.iterrows()):
             if i >= records_to_process:
                 break
-
             title = row['title'] if pd.notna(row['title']) else ""
             description = row['description'] if pd.notna(row['description']) else ""
-
             status.text(f"Analyzing {i+1}/{records_to_process}: {title[:30]}...")
-
             suggested_subject = get_huggingface_suggestions(title, description)
-
             if suggested_subject:
                 suggestions.append((title, suggested_subject))
-
             progress.progress((i + 1) / records_to_process)
-
+        
         status.empty()
-
+        progress.empty()
+        
         if suggestions:
             suggestions_df = pd.DataFrame(suggestions, columns=["Title", "Suggested Subject"])
-
-            # --- Display inside Dark Card ---
+            
+            # Apply custom CSS for table styling to match the others
             st.markdown("""
-                <div style='
-                    background-color: #2e2e2e;
-                    padding: 1.5rem;
-                    border-radius: 10px;
-                    margin-top: 1.5rem;
-                    color: lightgray;
-                '>
+            <style>
+            table {
+                width: 100%;
+                border-collapse: collapse;
+                color: #e0e0e0;
+            }
+            thead tr {
+                border-bottom: 1px solid #444;
+            }
+            th, td {
+                padding: 12px;
+                text-align: left;
+                border-bottom: 1px solid #444;
+            }
+            tr:hover {
+                background-color: #2c2c2c;
+            }
+            </style>
             """, unsafe_allow_html=True)
-
-            # --- Use st.table instead of st.dataframe for consistent dark styling ---
+            
+            # Use st.table for consistent styling
             st.table(suggestions_df.style
-                     .background_gradient(cmap="Greens", subset=["Suggested Subject"])
+                     .background_gradient(cmap="Greens", subset=["Suggested Subject"], vmin=0, vmax=1)
+                     .format({"Title": lambda x: x[:50] + "..." if len(x) > 50 else x})
                      .hide(axis="index")
+                     .set_properties(**{'text-align': 'left'})
             )
-
-            st.markdown("</div>", unsafe_allow_html=True)
-
         else:
             st.info("No metadata enhancement suggestions available.")
-
     else:
         st.info("Enable AI Suggestions to view recommendations.")
 else:
     st.success("All records already have subjects or no usable text available.")
 
+st.markdown("</div>", unsafe_allow_html=True)
+