Merge pull request #32 from Windsooon/add-threshold

add threshold feature

main.py

@@ -10,7 +10,7 @@
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s')
 
-def process_email(email_path, model):
+def process_email(email_path):
     '''
     Process the emails at the given path and return the results.
 
@@ -55,8 +55,8 @@ def export_to_csv(data, filename):
         for row in data:
             writer.writerow(row)
 
-def main(email_path, output_csv, model):
-    result = process_email(email_path, model)
+def main(email_path, output_csv):
+    result = process_email(email_path)
     if result:
         export_to_csv(result, output_csv)
         logging.info(f"Processed emails successfully and exported to CSV at {output_csv}.")

src/JobTracker/chatbot.py

@@ -130,8 +130,8 @@ def get_cost(self, mails):
 
     def gen_prompt(self, info):
         """Generate a prompt for the chatbot."""
-        return ('If this is a mail from a company I applied to or interviewed with before, '
-                'use get_mail_info to get information, here is the mail body: ' + info)
+        return (f'If this email has a {THRESHOLD}% confidence level that it is from a company I applied to or interviewed with before, '
+                f'use get_mail_info to get information, otherwise just ignore it. Here is the mail body: {info}')
 
     def get_content(self, info):
         """Get content from the chatbot based on the provided information."""

src/JobTracker/config.py

@@ -3,7 +3,7 @@
 MODEL = 'gpt-4-1106-preview'
 LLAMA_URL = "http://127.0.0.1:11434/api/generate"
 LLAMA_MODEL = "llama2"
-THRESHOLD = 30
+THRESHOLD = 80
 FIX_ANSWER = False
 
 PRICE = {