Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 from pypdf import PdfReader
 from resume_tools import create_resume_agent
-
+import google.generativeai as genai
+import os
 
 def read_pdf_file(file_path):
     """
@@ -24,7 +25,28 @@ def read_pdf_file(file_path):
 
 
 
-def process_inputs(resume, job_description, custom_instructions):
+def verify_gemini_api_key(api_key):
+    try:
+        genai.configure(api_key=api_key)
+        models = genai.list_models()
+        for model in models:
+            pass
+        return True,""
+    except Exception as e:
+        if "API key not valid" in str(e) or "PERMISSION_DENIED" in str(e):
+            return False,"Reason: The API key itself is likely incorrect or lacks necessary permissions."
+        elif "FAILED_PRECONDITION" in str(e) and "free tier is not available" in str(e):
+            return False,"Reason: Free tier might not be available in your region, or billing needs to be enabled."
+        elif "RESOURCE_EXHAUSTED" in str(e):
+            return  False,"Reason: Rate limit exceeded or quota exhausted for this API key."
+        return False,""
+    
+
+def process_inputs(resume, job_description, custom_instructions,gemini_api_key):
+    response,code = verify_gemini_api_key(gemini_api_key)
+    if not response:
+        return code
+    os.environ["GOOGLE_API_KEY"] = gemini_api_key
     if resume is not None:
         resume = read_pdf_file(resume)
     else:
@@ -131,7 +153,7 @@ def process_inputs(resume, job_description, custom_instructions):
     Do not generate final resume at once — output tool-wise only.
     """
     
-    print(prompt)
+    # print(prompt)
     return create_resume_agent(prompt)
 
 with gr.Blocks() as demo:
@@ -141,12 +163,13 @@ with gr.Blocks() as demo:
             resume = gr.File(label="Upload Resume", file_types=[".pdf"])
             job_description = gr.Textbox(label="Paste Job Description here", lines=5)
             custom_instructions = gr.Textbox(label="Just Instruct how you want it to be......!!!!", lines=5)
+            gemini_api_key = gr.Textbox(label="Enter your Gemini API key..!!! Don't worry we wont store it. You can get it from https://aistudio.google.com/app/apikey", lines=1)
             submit_btn = gr.Button("Submit")
         with gr.Column():
             output = gr.Textbox(label="LaTEX Resume File", lines=10)
     submit_btn.click(
         fn=process_inputs,
-        inputs=[resume, job_description,custom_instructions],
+        inputs=[resume, job_description,custom_instructions,gemini_api_key],
         outputs=output
     )