Update app.py

app.py

@@ -8,6 +8,9 @@ from openai import OpenAI
 import uuid
 import re
 import time
+import pandas as pd
+import matplotlib.pyplot as plt
+import seaborn as sns
 
 # Load environment variables from .env file
 load_dotenv()
@@ -157,13 +160,18 @@ def generate_answer(query, context):
         print(f"Error generating answer: {str(e)}")
         return f"Error generating answer: {str(e)}"
 
-def generate_hr_document(prompt):
-    print(f"Generating HR document with prompt: {prompt}")
+def generate_hr_document(document_type, additional_info):
+    print(f"Generating HR document: {document_type}")
     try:
+        prompt = f"""Generate a professional {document_type} for an HR department. 
+        Additional information: {additional_info}
+        
+        Important: Format the response as plain text, not markdown. Use appropriate line breaks and spacing for readability."""
+
         response = client.chat.completions.create(
-            model="gpt-4o-mini",  # Updated to use gpt-4o-mini
+            model="gpt-4o-mini",
             messages=[
-                {"role": "system", "content": "You are an HR assistant. Generate a professional HR document based on the given prompt."},
+                {"role": "system", "content": "You are an expert HR assistant. Generate a professional HR document based on the given type and additional information. Format the response as plain text, not markdown."},
                 {"role": "user", "content": prompt}
             ]
         )
@@ -172,11 +180,38 @@ def generate_hr_document(prompt):
         print(f"Error generating HR document: {str(e)}")
         return f"Error generating HR document: {str(e)}"
 
+def calculate_paye(annual_income):
+    tax_bands = [
+        (5880, 0),
+        (1320, 0.05),
+        (1560, 0.10),
+        (38000, 0.175),
+        (192000, 0.25),
+        (366240, 0.30),
+        (float('inf'), 0.35)
+    ]
+    
+    remaining_income = annual_income
+    total_tax = 0
+    
+    for band, rate in tax_bands:
+        if remaining_income <= 0:
+            break
+        taxable_amount = min(band, remaining_income)
+        tax = taxable_amount * rate
+        total_tax += tax
+        remaining_income -= taxable_amount
+    
+    return total_tax
+
+def calculate_ssnit(basic_salary):
+    return basic_salary * 0.055
+
 def main():
     st.set_page_config(page_title="HR Document Assistant", layout="wide")
     st.title("HR Document Assistant")
 
-    tab1, tab2, tab3, tab4 = st.tabs(["📤 Upload PDF", "🔍 Query Database", "📝 Generate HR Document", "🗑️ Clear Database"])
+    tab1, tab2, tab3, tab4, tab5 = st.tabs(["📤 Upload PDF", "🔍 Query Database", "📝 Generate HR Document", "🧮 Tax Calculator", "🗑️ Clear Database"])
 
     with tab1:
         st.header("Upload PDF")
@@ -201,13 +236,91 @@ def main():
 
     with tab3:
         st.header("Generate HR Document")
-        st.write("Enter a prompt to generate an HR document using GPT-4.")
-        prompt = st.text_area("Enter your prompt", placeholder="Describe the HR document you need...")
+        st.write("Select an HR document type and provide additional information to generate the document.")
+        
+        document_types = [
+            "Employment Contract", "Offer Letter", "Job Description", "Employee Handbook",
+            "Performance Review Form", "Disciplinary Action Form", "Leave Request Form",
+            "Onboarding Checklist", "Termination Letter", "Non-Disclosure Agreement (NDA)",
+            "Code of Conduct", "Workplace Policy", "Benefits Summary", "Compensation Plan",
+            "Training and Development Plan", "Resignation Letter", "Exit Interview Form",
+            "Employee Grievance Form", "Time-off Request Form", "Workplace Safety Guidelines"
+        ]
+        
+        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...")
+        
         if st.button("✍️ Generate Document"):
-            document = generate_hr_document(prompt)
-            st.text_area("Generated Document", value=document, height=400)
+            with st.spinner("Generating document..."):
+                document = generate_hr_document(selected_document, additional_info)
+            st.subheader(f"Generated {selected_document}")
+            st.text_area("Document Content", value=document, height=400)
+            st.download_button(
+                label="Download Document",
+                data=document,
+                file_name=f"{selected_document.lower().replace(' ', '_')}.txt",
+                mime="text/plain"
+            )
 
     with tab4:
+        st.header("Tax Calculator")
+        st.write("Calculate PAYE and SSNIT contributions based on annual income and basic salary.")
+        
+        salary_examples = {
+            "Entry Level": (36000, 30000),
+            "Mid Level": (72000, 60000),
+            "Senior Level": (120000, 90000),
+            "Executive": (240000, 180000)
+        }
+        
+        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)
+            basic_salary = st.number_input("Basic Salary (GH₵)", min_value=0.0, value=0.0, step=1000.0)
+        else:
+            annual_income, basic_salary = salary_examples[selected_example]
+            st.write(f"Annual Income: GH₵ {annual_income:.2f}")
+            st.write(f"Basic Salary: GH₵ {basic_salary:.2f}")
+        
+        if st.button("Calculate Taxes"):
+            ssnit_contribution = calculate_ssnit(basic_salary)
+            taxable_income = annual_income - ssnit_contribution
+            paye = calculate_paye(taxable_income)
+            net_income = annual_income - ssnit_contribution - paye
+            
+            col1, col2 = st.columns(2)
+            
+            with col1:
+                st.subheader("Tax Breakdown")
+                st.write(f"SSNIT Contribution: GH₵ {ssnit_contribution:.2f}")
+                st.write(f"PAYE: GH₵ {paye:.2f}")
+                st.write(f"Total Deductions: GH₵ {(ssnit_contribution + paye):.2f}")
+                st.write(f"Net Income: GH₵ {net_income:.2f}")
+            
+            with col2:
+                # Pie chart for income breakdown
+                fig, ax = plt.subplots(figsize=(3, 2))
+                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.axis('equal')
+                plt.title("Income Breakdown", fontsize=8)
+                st.pyplot(fig)
+            
+            # 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"],
+                "Rate (%)": [0, 5, 10, 17.5, 25, 30, 35]
+            }
+            df = pd.DataFrame(tax_data)
+            st.table(df)
+
+    with tab5:
         st.header("Clear Database")
         st.write("Use this option carefully. It will remove all data from the Pinecone index.")
         if st.button("🗑️ Clear Database"):