Update app.py

app.py

@@ -12,7 +12,7 @@ def main():
 
     if api_key:
         # Download BOE rates
-        download_boe_rates()
+        boe_rates_df = download_boe_rates()
 
         # Allow user to upload Excel sheet
         uploaded_file = st.file_uploader("Upload Excel file", type=["xlsx", "xls"])
@@ -40,7 +40,7 @@ def main():
                 })
 
                 # Calculate late interest
-                df_with_interest = calculate_late_interest(df_calculate, late_interest_rate)
+                df_with_interest = calculate_late_interest(df_calculate, late_interest_rate, boe_rates_df)
 
                 # Display calculated late interest
                 total_late_interest = df_with_interest['late_interest'].sum()
@@ -48,14 +48,7 @@ def main():
                 st.write(total_late_interest)
 
             # Generate conversation prompt
-            prompt = "I have analyzed the provided Excel sheet. "
-            if due_dates:
-                prompt += f"The due dates in the sheet are: {', '.join(str(date) for date in due_dates)}. "
-            if payment_dates:
-                prompt += f"The payment dates in the sheet are: {', '.join(str(date) for date in payment_dates)}. "
-            if amounts:
-                prompt += f"The amounts in the sheet are: {', '.join(str(amount) for amount in amounts)}. "
-            prompt += "Based on this information, what would you like to discuss?"
+            prompt = generate_conversation_prompt(df)
 
             # Allow user to engage in conversation
             user_input = st.text_input("Start a conversation:")
@@ -68,21 +61,36 @@ def main():
                         {"role": "system", "content": prompt},
                         {"role": "user", "content": user_input}
                     ],
-                    max_tokens=1800 
+                    max_tokens=1800  # Adjust this value to allow longer responses
                 )
                 response = completion.choices[0].message['content']
                 st.write("AI's Response:")
                 st.write(response)
+
+        # Button to clear cache
+        if st.button("Clear Cache"):
+            st.cache.clear()
+
     else:
         st.warning("Please enter your OpenAI API key.")
 
 # Function to calculate late interest
-def calculate_late_interest(data, late_interest_rate):
+def calculate_late_interest(data, late_interest_rate, boe_rates_df):
     # Calculate late days and late interest
     data['late_days'] = (data['payment_date'] - data['due_date']).dt.days.clip(lower=0)
     data['late_interest'] = data['late_days'] * data['amount'] * (late_interest_rate / 100)
+
+    # Consider additional factors like Bank of England base rate
+    data['boe_base_rate'] = data['due_date'].map(lambda x: get_boe_base_rate(x, boe_rates_df))
+    data['late_interest'] += data['amount'] * (data['boe_base_rate'] / 100)
+
     return data
 
+# Function to get Bank of England base rate for a given date
+def get_boe_base_rate(date, boe_rates_df):
+    closest_date = boe_rates_df['Date Changed'].iloc[(boe_rates_df['Date Changed']-date).abs().argsort()[0]]
+    return boe_rates_df[boe_rates_df['Date Changed'] == closest_date]['Base Rate'].values[0]
+
 # Function to analyze Excel sheet and extract relevant information
 def analyze_excel(df):
     # Extract due dates and payment dates
@@ -111,10 +119,26 @@ def download_boe_rates():
             df = pd.read_html(response.text)[0]
             df.to_csv('boe_rates.csv', index=False)
             st.success("Bank of England rates downloaded successfully.")
+            return df
         else:
             st.error("Failed to retrieve data from the Bank of England website.")
+            return None
     except requests.RequestException as e:
         st.error(f"Failed to download rates: {e}")
+        return None
+
+# Function to generate conversation prompt
+def generate_conversation_prompt(df):
+    prompt = "I have analyzed the provided Excel sheet. "
+    due_dates, payment_dates, amounts = analyze_excel(df)
+    if due_dates:
+        prompt += f"The due dates in the sheet are: {', '.join(str(date) for date in due_dates)}. "
+    if payment_dates:
+        prompt += f"The payment dates in the sheet are: {', '.join(str(date) for date in payment_dates)}. "
+    if amounts:
+        prompt += f"The amounts in the sheet are: {', '.join(str(amount) for amount in amounts)}. "
+    prompt += "Based on this information, what would you like to discuss?"
+    return prompt
 
 if __name__ == "__main__":
     main()