All files

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import openai
 from utils.constants import model_family_mapping, model_name_mapping
 from utils.utils import PitchPerfect, pdf_loader
 
@@ -41,16 +42,20 @@ with st.sidebar:
         token = st.text_input("OpenAI API Key", type="password", key="openai_key")
     else:
         token = st.text_input("Hugging Face Token", type="password", key="hf_token")
-        
-    if st.button("Initialize with the provided keys"):
-        try:
-            st.session_state.pitch_perfect = PitchPerfect(model = model_name, model_family = model_family, token = token)
-            st.session_state.api_configured = True
-            st.success("Successfully configured the API clients with provided keys!")
-        
-        except Exception as e:
-            st.error(f"Error initializing API clients: {str(e)}")
-            st.session_state.api_configured = False
+    
+    if token != "":
+        if st.button("Initialize with the provided keys"):
+            try:
+                st.session_state.pitch_perfect = PitchPerfect(model = model_name, model_family = model_family, token = token)
+                if st.session_state.pitch_perfect.client == "INVALID":
+                    st.error(st.session_state.pitch_perfect.error)
+                else:
+                    st.session_state.api_configured = True
+                    st.success("Successfully configured the API clients with provided keys!")
+            
+            except Exception as e:
+                st.error(f"Error initializing API clients: {str(e)}")
+                st.session_state.api_configured = False
             
     if st.session_state.api_configured:
         upload_cv = st.file_uploader("Upload CV in PDF format", type=["pdf"])

utils/utils.py

@@ -1,5 +1,7 @@
 from langchain_community.document_loaders import PyPDFLoader
 from huggingface_hub import InferenceClient
+import requests
+import openai
 from openai import OpenAI
 from utils.constants import system_prompt
 
@@ -22,12 +24,61 @@ class PitchPerfect:
         
         if model_family == "gpt":
             self.client = OpenAI(api_key = token)
+            self.check_openai_token()
         elif model_family == "together":
             self.client = InferenceClient(provider = "together", api_key = token)
+            self.check_hf_token(token)
         else:
             self.client = InferenceClient(provider="hf-inference", api_key = token)
+            self.check_hf_token(token)
             
-        
+    def check_openai_token(self):
+        try:
+            self.client.models.list()
+        except openai.AuthenticationError:
+            print("❌ Invalid API key.")
+            self.client = "INVALID"
+            self.error = "❌ Invalid API key."
+        except Exception as e:
+            print(f"❌ An error occurred: {e}")
+            self.client = "INVALID"
+            self.error = "❌ An error occurred: {e}"
+            
+    def check_hf_token(self, token):
+        headers = {"Authorization": f"Bearer {token}"}
+        try:
+            response = requests.get("https://huggingface.co/api/whoami", headers=headers)
+            if response.status_code == 200:
+                user_info = response.json()
+                print(f"✅ Token is valid. Logged in as: {user_info.get('name', 'Unknown User')}")
+            elif response.status_code == 401:
+                print("❌ Invalid Hugging Face token.")
+                self.client = "INVALID"
+                self.error = "❌ Invalid HF Token."
+            elif response.status_code == 403:
+                print("❌ Token does not have the necessary permissions.")
+                self.client = "INVALID"
+                self.error = "❌ Token does not have the necessary permissions."
+            else:
+                print(f"❌ Unexpected error: {response.status_code} - {response.text}")
+                self.client = "INVALID"
+                self.error = f"❌ Unexpected error: {response.status_code} - {response.text}"
+        except Exception as e:
+            print(f"❌ An unexpected error occurred: {e}")
+            self.client = "INVALID"
+            self.error = f"❌ An unexpected error occurred: {e}"
+            
+    # def check_api_key(self):
+    #     try:
+    #         # Make a simple request to test the API key
+    #         self.client.models.list()
+    #         print("✅ API key is valid.")
+    #     except openai.APIConnectionError:
+    #         print("❌ Network error. Please check your connection.")
+    #     except openai.AuthenticationError:
+    #         print("❌ Invalid API key.")
+    #     except Exception as e:
+    #         print(f"❌ An error occurred: {e}")
             
     def prepare_user_prompt(self, job_title, company, job_desc, cv_data, word_limit):