Upload 11 files

app_hug.py

@@ -0,0 +1,487 @@
+from fastapi import FastAPI, HTTPException, BackgroundTasks
+from pydantic import BaseModel, EmailStr, Field
+from typing import Optional, Tuple
+from enum import Enum
+import os
+import base64
+import pickle
+import pandas as pd
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+from langchain.schema import HumanMessage, SystemMessage
+from email.mime.text import MIMEText
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+from datetime import datetime
+import json
+
+# Load environment variables (not needed on Hugging Face, but harmless)
+load_dotenv()
+
+# ------------------------------------------
+# Helper: Write GOOGLE_CREDENTIALS_JSON to file if needed
+# ------------------------------------------
+def ensure_credentials_file():
+    credentials_env = os.getenv("GOOGLE_CREDENTIALS_JSON")
+    credentials_path = "credentials_SYNAPSE.json"
+    if not os.path.exists(credentials_path):
+        if not credentials_env:
+            raise Exception("GOOGLE_CREDENTIALS_JSON not found in environment variables.")
+        try:
+            parsed_json = json.loads(credentials_env)
+        except json.JSONDecodeError:
+            raise Exception("Invalid JSON in GOOGLE_CREDENTIALS_JSON")
+        with open(credentials_path, "w") as f:
+            json.dump(parsed_json, f, indent=2)
+    return credentials_path
+
+# ------------------------------------------
+# FastAPI app
+# ------------------------------------------
+app = FastAPI(title="Recruitment Message Generator API", version="1.0.0")
+
+SCOPES = ["https://www.googleapis.com/auth/gmail.send"]
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+# ------------------------------------------
+# Enums and Models
+# ------------------------------------------
+class MessageType(str, Enum):
+    OUTREACH = "outreach"
+    INTRODUCTORY = "introductory"
+    FOLLOWUP = "followup"
+
+class GenerateMessageRequest(BaseModel):
+    job_evaluation: str
+    sender_email: EmailStr
+    recruiter_email: Optional[EmailStr] = None
+    recipient_email: EmailStr
+    candidate_name: str
+    current_role: str
+    current_company: str
+    company_name: str
+    role: str
+    recruiter_name: str
+    organisation: str
+    message_type: MessageType
+    send_email: bool = False
+
+class FeedbackRequest(BaseModel):
+    message: str
+    feedback: str
+
+class AuthenticateRequest(BaseModel):
+    email: EmailStr
+
+class AuthenticateResponse(BaseModel):
+    success: bool
+    message: str
+    error: Optional[str] = None
+
+class MessageResponse(BaseModel):
+    success: bool
+    message: str
+    email_sent: bool = False
+    email_subject: Optional[str] = None
+    error: Optional[str] = None
+
+# ------------------------------------------
+# Gmail Helper Functions
+# ------------------------------------------
+def get_token_file_path(email: str) -> str:
+    tokens_dir = "gmail_tokens"
+    if not os.path.exists(tokens_dir):
+        os.makedirs(tokens_dir)
+    safe_email = email.replace("@", "_at_").replace(".", "_dot_")
+    return os.path.join(tokens_dir, f"token_{safe_email}.pickle")
+
+def check_user_token_exists(email: str) -> bool:
+    token_file = get_token_file_path(email)
+    return os.path.exists(token_file)
+
+def load_user_credentials(email: str):
+    token_file = get_token_file_path(email)
+    if os.path.exists(token_file):
+        try:
+            with open(token_file, 'rb') as token:
+                creds = pickle.load(token)
+                return creds
+        except Exception:
+            if os.path.exists(token_file):
+                os.remove(token_file)
+    return None
+
+def save_user_credentials(email: str, creds):
+    token_file = get_token_file_path(email)
+    with open(token_file, 'wb') as token:
+        pickle.dump(creds, token)
+
+def create_new_credentials(email: str):
+    credentials_path = ensure_credentials_file()
+    flow = InstalledAppFlow.from_client_secrets_file(
+        credentials_path, SCOPES
+    )
+    creds = flow.run_local_server(port=0)
+    save_user_credentials(email, creds)
+    return creds
+
+def authenticate_gmail(email: str, create_if_missing: bool = False):
+    creds = load_user_credentials(email)
+    if creds:
+        if creds.expired and creds.refresh_token:
+            try:
+                creds.refresh(Request())
+                save_user_credentials(email, creds)
+            except Exception:
+                if create_if_missing:
+                    try:
+                        creds = create_new_credentials(email)
+                    except:
+                        return None
+                else:
+                    return None
+        elif not creds.valid:
+            creds = None
+    if not creds:
+        if create_if_missing:
+            try:
+                creds = create_new_credentials(email)
+            except:
+                return None
+        else:
+            return None
+    try:
+        service = build("gmail", "v1", credentials=creds)
+        return service
+    except Exception:
+        return None
+
+def create_email_message(sender: str, to: str, subject: str, message_text: str, reply_to: Optional[str] = None):
+    message = MIMEText(message_text)
+    message["to"] = to
+    message["from"] = sender
+    message["subject"] = subject
+    if reply_to:
+        message["reply-to"] = reply_to
+    raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode()
+    return {"raw": raw_message}
+
+def send_gmail_message(service, user_id: str, message: dict):
+    try:
+        result = service.users().messages().send(userId=user_id, body=message).execute()
+        return result is not None
+    except HttpError:
+        return False
+
+# ------------------------------------------
+# LLM (OpenAI) Message Generation Helpers
+# ------------------------------------------
+def refine_message_with_feedback(
+    original_message: str,
+    feedback: str,
+) -> Tuple[str, str]:
+    api_key = os.getenv("OPENAI_API_KEY")
+    llm = ChatOpenAI(
+        model="gpt-4o-mini",
+        temperature=0.7,
+        max_tokens=600,
+        openai_api_key=api_key
+    )
+    prompt = f"""
+    Please refine the following recruitment message based on the provided feedback:
+
+    ORIGINAL MESSAGE:
+    {original_message}
+
+    FEEDBACK:
+    {feedback}
+
+    Please provide your response in the following format:
+    SUBJECT: [Your subject line here]
+
+    BODY:
+    [Your refined email body content here]
+    Keep the same tone and intent as the original message, but incorporate the feedback to improve it.
+    """
+    try:
+        messages = [
+            SystemMessage(content="You are a professional recruitment message writer. Refine the given message based on feedback while maintaining professionalism and the original intent."),
+            HumanMessage(content=prompt)
+        ]
+        response = llm.invoke(messages)
+        content = response.content.strip()
+        subject_line = ""
+        body_content = ""
+        lines = content.split('\n')
+        body_found = False
+        body_lines = []
+        for line in lines:
+            if line.strip().startswith('SUBJECT:'):
+                subject_line = line.replace('SUBJECT:', '').strip()
+            elif line.strip().startswith('BODY:'):
+                body_found = True
+            elif body_found and line.strip():
+                body_lines.append(line)
+        body_content = '\n'.join(body_lines).strip()
+        if not subject_line:
+            subject_line = "Recruitment Opportunity - Updated"
+        if not body_content:
+            body_content = content
+        return subject_line, body_content
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error refining message: {str(e)}")
+
+def generate_recruitment_message_with_subject(
+    msg_type: str,
+    company: str,
+    role_title: str,
+    recruiter: str,
+    org: str,
+    candidate: str,
+    current_pos: str,
+    evaluation: str,
+    feedback: Optional[str] = None
+) -> Tuple[str, str]:
+    api_key = os.getenv("OPENAI_API_KEY")
+    llm = ChatOpenAI(
+        model="gpt-4o-mini",
+        temperature=0.7,
+        max_tokens=600,
+        openai_api_key=api_key
+    )
+    base_prompt = f"""
+    Generate a professional recruitment {msg_type} with the following details:
+    - Company hiring: {company}
+    - Role: {role_title}
+    - Recruiter: {recruiter} from {org}
+    - Candidate: {candidate}
+    - Candidate's current position: {current_pos}
+    - Evaluation: {evaluation}
+    """
+    if msg_type == "outreach":
+        prompt = base_prompt + """
+        Create an initial outreach message that:
+        - Introduces the recruiter and organization
+        - Mentions the specific role and company
+        - Expresses interest in discussing the opportunity
+        - Keeps it short and to the point.
+        """
+    elif msg_type == "introductory":
+        prompt = base_prompt + """
+        Create an introductory message that:
+        - Thanks the candidate for their initial response
+        - Provides more details about the role and company
+        - Explains why this opportunity aligns with their background
+        - Suggests next steps (like a call or meeting)
+        - Maintains a warm, professional tone
+        """
+    else:  # followup
+        prompt = base_prompt + """
+        Create a follow-up message that:
+        - References previous communication
+        - Reiterates interest in the candidate
+        - Addresses any potential concerns
+        - Provides additional compelling reasons to consider the role
+        - Includes a clear call to action
+        """
+    if feedback:
+        prompt += f"\n\nPlease modify the message based on this feedback: {feedback}"
+    prompt += """
+
+    Please provide your response in the following format:
+    SUBJECT: [Your subject line here]
+
+    BODY:
+    [Your email body content here]
+    """
+    try:
+        messages = [
+            SystemMessage(content="You are a professional recruitment message writer. Generate both an email subject line and body content. Follow the exact format requested."),
+            HumanMessage(content=prompt)
+        ]
+        response = llm.invoke(messages)
+        content = response.content.strip()
+        subject_line = ""
+        body_content = ""
+        lines = content.split('\n')
+        body_found = False
+        body_lines = []
+        for line in lines:
+            if line.strip().startswith('SUBJECT:'):
+                subject_line = line.replace('SUBJECT:', '').strip()
+            elif line.strip().startswith('BODY:'):
+                body_found = True
+            elif body_found and line.strip():
+                body_lines.append(line)
+        body_content = '\n'.join(body_lines).strip()
+        if not subject_line:
+            subject_line = f"Opportunity at {company} - {role_title}"
+        if not body_content:
+            body_content = content
+        return subject_line, body_content
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error generating message: {str(e)}")
+
+# ------------------------------------------
+# FastAPI Endpoints
+# ------------------------------------------
+@app.get("/")
+async def root():
+    return {
+        "message": "Recruitment Message Generator API",
+        "version": "1.0.0",
+        "endpoints": [
+            "/generate-message",
+            "/refine-message",
+            "/authenticate",
+            "/docs"
+        ]
+    }
+
+@app.post("/generate-message", response_model=MessageResponse)
+async def generate_message(request: GenerateMessageRequest, background_tasks: BackgroundTasks):
+    try:
+        current_position = f"{request.current_role} at {request.current_company}"
+        email_subject, generated_message = generate_recruitment_message_with_subject(
+            msg_type=request.message_type.value.replace('followup', 'follow-up'),
+            company=request.company_name,
+            role_title=request.role,
+            recruiter=request.recruiter_name,
+            org=request.organisation,
+            candidate=request.candidate_name,
+            current_pos=current_position,
+            evaluation=request.job_evaluation
+        )
+        email_sent = False
+        if request.send_email:
+            registered_users = []
+            if os.path.exists("registered_users.csv"):
+                df = pd.read_csv("registered_users.csv")
+                registered_users = df['email'].tolist() if 'email' in df.columns else []
+            if request.sender_email.lower() not in [user.lower() for user in registered_users]:
+                return MessageResponse(
+                    success=True,
+                    message=generated_message,
+                    email_sent=False,
+                    email_subject=email_subject,
+                    error="Email not sent: Sender email is not registered"
+                )
+            service = authenticate_gmail(request.sender_email)
+            if service:
+                email_message = create_email_message(
+                    sender=request.sender_email,
+                    to=request.recipient_email,
+                    subject=email_subject,
+                    message_text=generated_message,
+                    reply_to=request.recruiter_email
+                )
+                email_sent = send_gmail_message(service, "me", email_message)
+                if not email_sent:
+                    return MessageResponse(
+                        success=True,
+                        message=generated_message,
+                        email_sent=False,
+                        email_subject=email_subject,
+                        error="Email not sent: Failed to send via Gmail API"
+                    )
+            else:
+                return MessageResponse(
+                    success=True,
+                    message=generated_message,
+                    email_sent=False,
+                    email_subject=email_subject,
+                    error="Email not sent: Gmail authentication failed"
+                )
+        return MessageResponse(
+            success=True,
+            message=generated_message,
+            email_sent=email_sent,
+            email_subject=email_subject
+        )
+    except Exception as e:
+        return MessageResponse(
+            success=False,
+            message="",
+            error=str(e)
+        )
+
+@app.post("/refine-message", response_model=MessageResponse)
+async def refine_message(request: FeedbackRequest):
+    try:
+        email_subject, refined_message = refine_message_with_feedback(
+            original_message=request.message,
+            feedback=request.feedback
+        )
+        return MessageResponse(
+            success=True,
+            message=refined_message,
+            email_sent=False,
+            email_subject=email_subject
+        )
+    except Exception as e:
+        return MessageResponse(
+            success=False,
+            message="",
+            error=str(e)
+        )
+
+@app.post("/authenticate", response_model=AuthenticateResponse)
+async def authenticate_user(request: AuthenticateRequest):
+    try:
+        if check_user_token_exists(request.email):
+            service = authenticate_gmail(request.email, create_if_missing=False)
+            if service:
+                return AuthenticateResponse(
+                    success=True,
+                    message="User already authenticated and token is valid"
+                )
+            else:
+                service = authenticate_gmail(request.email, create_if_missing=True)
+                if service:
+                    return AuthenticateResponse(
+                        success=True,
+                        message="Token refreshed successfully"
+                    )
+                else:
+                    return AuthenticateResponse(
+                        success=False,
+                        message="Failed to refresh token",
+                        error="Could not refresh existing token. Please check credentials.json"
+                    )
+        else:
+            try:
+                creds = create_new_credentials(request.email)
+                if creds:
+                    return AuthenticateResponse(
+                        success=True,
+                        message="Authentication successful. Token created and saved."
+                    )
+                else:
+                    return AuthenticateResponse(
+                        success=False,
+                        message="Authentication failed",
+                        error="Failed to create credentials"
+                    )
+            except Exception as e:
+                return AuthenticateResponse(
+                    success=False,
+                    message="Authentication failed",
+                    error=str(e)
+                )
+    except Exception as e:
+        return AuthenticateResponse(
+            success=False,
+            message="Authentication error",
+            error=str(e)
+        )
+
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy", "timestamp": datetime.utcnow().isoformat()}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)

gmail_tokens.csv

@@ -0,0 +1,5 @@
+email,access_token,refresh_token,token_uri,client_id,client_secret,scopes,expiry,token
+[email protected],ya29.a0AS3H6NwggjTEMH3RZ09JZnA_34k918mSE1hemVWoc3VtmiZf3h2AcrHJGmYNov6VNO7HSnouMe2XOVNY5jn3xmH0dCytDnRJV9fcu1WfiLHwBjLND2B8xK7KPPQ65WrwIRqG7uCmnt-4eQgNS20-5NMKoUy62qruYbR5xg5vaCgYKARISARYSFQHGX2MifkdYHkS26F8ymU9Plo40kA0175,1//0gdA8tU6VFZQgCgYIARAAGBASNwF-L9IrHULnmrqCQ0cm8T9b_823Tm0cvoOE5u-6_kIOSKR3vArNsPhOqoDWcwjhNEAMVCZf_a4,https://oauth2.googleapis.com/token,134246740312-hq134ss12i47b5tdqa6v9dvmpb34el7u.apps.googleusercontent.com,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,https://www.googleapis.com/auth/gmail.send,2025-06-30T06:48:02.130965,gmail_tokens\token_ali_at_getjobsbysynapse_dot_com.pickle
+[email protected],ya29.a0AS3H6NyUUZzfX6k3YLaze3_oCfYlYO1NWBQoS5tL-lZMyPMGw-Bda36wtGuecucSgv3YV8nP-zJjnkr-CTtVz2advTmFGkWqoiKRwMbP3ds8Rc__sLhXXJyGocgodeG8_ZELZ7sbMu1yxE86wahIkUakPxDXTgKerIHsDUtIaCgYKAQ8SARASFQHGX2MiwN7pFdWL_KPNLT9ifc7GCw0175,1//0gvpymvUiojGZCgYIARAAGBASNwF-L9Ird8_yKPv3HWCxT9dhpWLR_KwYlBPOMIpiCcQnwhDqi6ac8-4oLhDPA6T9xNPdPy9URq4,https://oauth2.googleapis.com/token,134246740312-hq134ss12i47b5tdqa6v9dvmpb34el7u.apps.googleusercontent.com,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,https://www.googleapis.com/auth/gmail.send,2025-06-30T06:50:47.931799,gmail_tokens\token_ali_at_matchwithsynapse_dot_com.pickle
+[email protected],ya29.a0AS3H6NyPysZGsyDxwn9jo5zHkwh1I2wycyfD8gjw655YGqslNR37j5WhFXewjT1JU5NFh1NbZCh-SeWuCOZp-_ZvGu7elsswsLg7_YMYx38XdNwnu-cVwdGhKT6TakikmCQMF7ATM8w_ykAbT75VX3Wz38Ho9pxrdsZWQVSkaCgYKAS4SARMSFQHGX2MigGQZccNrOEfc4Hr_8IPaJQ0175,1//0gyVDLQU8kisCCgYIARAAGBASNwF-L9IrNQZg0yHD_VnM3bU14X4ZlWEyY6fUqa5ltbnWW_j08egtAfNDw4AGkYvSwByoa_sv7Xw,https://oauth2.googleapis.com/token,134246740312-hq134ss12i47b5tdqa6v9dvmpb34el7u.apps.googleusercontent.com,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,https://www.googleapis.com/auth/gmail.send,2025-06-30T06:51:40.576934,gmail_tokens\token_ali_dot_tag_at_elitesrn_dot_com.pickle
+[email protected],ya29.a0AS3H6NzBJGynA2pYXHOkfSU5S6_FnOLn6lu2jLg7gneTlwbxHGWcaXPtp6JFdATR2CSZmC8qHzyzTyRhDwqI1BUvXQqDdpgKTV7Px2l4bWXPCZzimWk-DbSIxBXolvTnrlDXWq5KsBV3sPkry6SB3v2nNaU-34MnFB1VdvGrwQaCgYKAdASARcSFQHGX2MiRaUEVWxUgmV_Pdcasc74-A0177,1//0gcLQ1zVYdP7yCgYIARAAGBASNwF-L9Ir9fu9m_IljS2LAxdVEnjEAxBzM4CCC7voPLqpUKlT9zHnzustS7fbx9cDq2QGuPFCbjI,https://oauth2.googleapis.com/token,667261841954-q5gdoc76bvcn3l72q993t2b62tfk8pvq.apps.googleusercontent.com,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,https://www.googleapis.com/auth/gmail.send,2025-06-30T06:58:01.533285,gmail_tokens\token_kapooraman2468_at_gmail_dot_com.pickle

gmail_tokens/token_ali_at_getjobsbysynapse_dot_com.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:09059e2904e3d5beed937b5fb1edfe78a77d15ce04905d50cb78b7f2cc52efbf
+size 999

gmail_tokens/token_ali_at_matchwithsynapse_dot_com.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b3d012ca1145c9a4f1400f3187f4b7c82b3d5f2cb5e4d392937865635fbd535
+size 999

gmail_tokens/token_ali_dot_tag_at_elitesrn_dot_com.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd0a5f5fd7484649823ef686bd91f0863f356d363b46a23dd736428033ff8f9e
+size 999

gmail_tokens/token_ali_dot_taghikhani_at_getjobsbysynapse_dot_com.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25082999b045e7be6d0a797766fef72b72e6a4df668d5087ad4ef95644b8d5da
+size 999

gmail_tokens/token_aman_at_synapseint_dot_com.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb3dd4b977b6ba88a47b8791490404dee73e80abeffe48108a7cc1a754261f6c
+size 999

gmail_tokens/token_aman_dot_21ug3019_at_iiitranchi_dot_ac_dot_in.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:04b2555649ba9cb3282cc86c061691229a7bb37ad53b60df0992133757dd1aea
+size 999

gmail_tokens/token_kapooraman2468_at_gmail_dot_com.pickle

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ebbba500ecaf05629d824c02dfdfec628f4e16d93cd6e50891a4eeb931eee991
+size 1001

registered_users.csv

@@ -0,0 +1,10 @@
+email
+[email protected]
+[email protected]
+[email protected]
+[email protected]
+[email protected]
+[email protected]
+[email protected]
+[email protected]
+[email protected]

requirements.txt

@@ -0,0 +1,14 @@
+fastapi
+google_api_python_client
+google_auth_oauthlib
+langchain
+langchain_openai
+pandas
+protobuf
+psycopg2
+psycopg2_binary
+pydantic
+python-dotenv
+SQLAlchemy
+streamlit
+uvicorn