Update app.py

app.py

@@ -464,16 +464,6 @@ 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 = []
@@ -498,41 +488,58 @@ if not incomplete_with_desc.empty:
         if suggestions:
             suggestions_df = pd.DataFrame(suggestions, columns=["Title", "Suggested Subject"])
             
-            # Apply custom CSS for table styling to match the others
-            st.markdown("""
-            <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)
+            # Create a custom dark-styled HTML table instead
+            html_table = """
+            <div style="background-color: #1e1e1e; padding: 1.5rem; border-radius: 10px; margin-top: 1rem;">
+                <table style="width: 100%; border-collapse: collapse; color: #e0e0e0;">
+                    <thead>
+                        <tr style="border-bottom: 1px solid #444;">
+                            <th style="padding: 12px; text-align: left; color: #e0e0e0;">Title</th>
+                            <th style="padding: 12px; text-align: left; color: #e0e0e0;">Suggested Subject</th>
+                        </tr>
+                    </thead>
+                    <tbody>
+            """
             
-            # Use st.table for consistent styling
-            st.table(suggestions_df.style
-                     .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'})
-            )
+            for _, row in suggestions_df.iterrows():
+                title = row['Title']
+                title_display = title[:50] + "..." if len(title) > 50 else title
+                subject = row['Suggested Subject']
+                
+                # Calculate a shade of green based on confidence or some other metric
+                # For demonstration, using a fixed green shade
+                green_shade = "rgba(0, 100, 0, 0.3)"
+                
+                html_table += f"""
+                    <tr style="border-bottom: 1px solid #444;">
+                        <td style="padding: 12px; text-align: left;">{title_display}</td>
+                        <td style="padding: 12px; text-align: left; background-color: {green_shade};">{subject}</td>
+                    </tr>
+                """
+            
+            html_table += """
+                    </tbody>
+                </table>
+            </div>
+            """
+            
+            st.markdown(html_table, unsafe_allow_html=True)
         else:
-            st.info("No metadata enhancement suggestions available.")
+            st.markdown("""
+                <div style="background-color: #1e1e1e; padding: 1.5rem; border-radius: 10px; margin-top: 1rem; color: #e0e0e0;">
+                    No metadata enhancement suggestions available.
+                </div>
+            """, unsafe_allow_html=True)
     else:
-        st.info("Enable AI Suggestions to view recommendations.")
+        st.markdown("""
+            <div style="background-color: #1e1e1e; padding: 1.5rem; border-radius: 10px; margin-top: 1rem; color: #e0e0e0;">
+                Enable AI Suggestions to view recommendations.
+            </div>
+        """, unsafe_allow_html=True)
 else:
-    st.success("All records already have subjects or no usable text available.")
-
-st.markdown("</div>", unsafe_allow_html=True)
+    st.markdown("""
+        <div style="background-color: #1e1e1e; padding: 1.5rem; border-radius: 10px; margin-top: 1rem; color: #e0e0e0;">
+            All records already have subjects or no usable text available.
+        </div>
+    """, unsafe_allow_html=True)