Update app.py

app.py

@@ -209,9 +209,21 @@ def calculate_ssnit(basic_salary):
 
 def main():
     st.set_page_config(page_title="HR Document Assistant", layout="wide")
-    st.title("HR Document Assistant")
-
-    tab1, tab2, tab3, tab4, tab5 = st.tabs(["📤 Upload PDF", "🔍 Query Database", "📝 Generate HR Document", "🧮 Tax Calculator", "🗑️ Clear Database"])
+    
+    # Create a header with logo and title
+    col1, col2 = st.columns([1, 4])
+    with col1:
+        st.image("logo.png", width=200)  # Adjust the width as needed
+    with col2:
+        st.title("HR Document Assistant")
+    
+    tab1, tab2, tab3, tab4, tab5 = st.tabs([
+        "📤 Upload PDF", 
+        "🔍 Query Database", 
+        "📝 Generate HR Document", 
+        "🧮 Tax Calculator", 
+        "🗑️ Clear Database"
+    ])
 
     with tab1:
         st.header("Upload PDF")
@@ -248,8 +260,10 @@ def main():
         ]
         
         selected_document = st.selectbox("Select HR Document Type", document_types)
-        additional_info = st.text_area("Additional Information", 
-                                       placeholder="Enter any specific details or requirements for the document...")
+        additional_info = st.text_area(
+            "Additional Information", 
+            placeholder="Enter any specific details or requirements for the document..."
+        )
         
         if st.button("✍️ Generate Document"):
             with st.spinner("Generating document..."):
@@ -274,8 +288,10 @@ def main():
             "Executive": (240000, 180000)
         }
         
-        selected_example = st.selectbox("Select a salary example or enter custom values:", 
-                                        ["Custom"] + list(salary_examples.keys()))
+        selected_example = st.selectbox(
+            "Select a salary example or enter custom values:", 
+            ["Custom"] + list(salary_examples.keys())
+        )
         
         if selected_example == "Custom":
             annual_income = st.number_input("Annual Income (GH₵)", min_value=0.0, value=0.0, step=1000.0)
@@ -306,7 +322,10 @@ def main():
                 sizes = [ssnit_contribution, paye, net_income]
                 labels = ['SSNIT', 'PAYE', 'Net']
                 colors = ['#ff9999', '#66b3ff', '#99ff99']
-                ax.pie(sizes, labels=labels, colors=colors, autopct='%1.1f%%', startangle=90, textprops={'fontsize': 6})
+                ax.pie(
+                    sizes, labels=labels, colors=colors, 
+                    autopct='%1.1f%%', startangle=90, textprops={'fontsize': 6}
+                )
                 ax.axis('equal')
                 plt.title("Income Breakdown", fontsize=8)
                 st.pyplot(fig)
@@ -314,7 +333,11 @@ def main():
             # Display tax rates by income bracket as a table
             st.subheader("Tax Rates by Income Bracket")
             tax_data = {
-                "Income Range (GH₵)": ["0 - 5,880", "5,881 - 7,200", "7,201 - 8,760", "8,761 - 46,760", "46,761 - 238,760", "238,761 - 605,000", "Above 605,000"],
+                "Income Range (GH₵)": [
+                    "0 - 5,880", "5,881 - 7,200", "7,201 - 8,760", 
+                    "8,761 - 46,760", "46,761 - 238,760", 
+                    "238,761 - 605,000", "Above 605,000"
+                ],
                 "Rate (%)": [0, 5, 10, 17.5, 25, 30, 35]
             }
             df = pd.DataFrame(tax_data)