Spaces:
Sleeping
Sleeping
| import gspread | |
| import os | |
| import json | |
| import pandas | |
| import streamlit as st | |
| from google.oauth2.service_account import Credentials | |
| from datetime import datetime | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| app_pswd = os.environ.get("app_pswd") | |
| sheet_id = os.environ.get("sheet_id") | |
| email_add = "[email protected]" | |
| my_name = "harsh sethi" | |
| template_col_name = "template" | |
| def customerDetailsGsheets( objective,personal_details,industry,recipientRole,recipientName): | |
| scopes = [ | |
| "https://www.googleapis.com/auth/spreadsheets" | |
| ] | |
| # Load the credentials from the environment variable | |
| json_string = os.getenv('credentials') | |
| if not json_string: | |
| return {"response": "Credentials not found in environment variable.", "status": "error"} | |
| # Convert the string back to a JSON object | |
| try: | |
| credentials = json.loads(json_string) | |
| except json.JSONDecodeError: | |
| return {"response": "Failed to decode JSON credentials.", "status": "error"} | |
| # Create credentials from the JSON object | |
| creds = Credentials.from_service_account_info(credentials, scopes=scopes) | |
| client = gspread.authorize(creds) | |
| # The sheet ID should also be stored in a secret or defined | |
| sheet_id = os.getenv('sheet_id') # Example of getting sheet_id from environment variable | |
| try: | |
| workbook = client.open_by_key(sheet_id) | |
| except gspread.SpreadsheetNotFound: | |
| return {"response": "Spreadsheet not found.", "status": "error"} | |
| # Select the specific worksheet (tab) where you want to append the data | |
| try: | |
| sheet = workbook.worksheet("Sheet1") # Add the worksheet name here | |
| except gspread.WorksheetNotFound: | |
| return {"response": "Worksheet not found.", "status": "error"} | |
| # Append the customer details as a new row in the worksheet | |
| if not (objective and personal_details): | |
| return { | |
| "response": "Please enter the important details" | |
| } | |
| try: | |
| sheet.append_row([objective, personal_details, industry, recipientRole, recipientName]) | |
| return { | |
| "response": "Saved customer details in Google Sheet.", | |
| "display": { | |
| "text": f"Customer details saved in Google Sheet: [Link]({workbook.url})" | |
| }, | |
| "status": "success" | |
| } | |
| except Exception as e: | |
| return {"response": f"Error saving data: {str(e)}", "status": "error"} | |
| def saveTemplate(template:str): | |
| scopes = [ | |
| "https://www.googleapis.com/auth/spreadsheets" | |
| ] | |
| json_string = os.getenv('credentials') | |
| if not json_string: | |
| return {"response": "Credentials not found in environment variable.", "status": "error"} | |
| # Convert the string back to a JSON object | |
| try: | |
| credentials = json.loads(json_string) | |
| except json.JSONDecodeError: | |
| return {"response": "Failed to decode JSON credentials.", "status": "error"} | |
| # Create credentials from the JSON object | |
| creds = Credentials.from_service_account_info(credentials, scopes=scopes) | |
| client = gspread.authorize(creds) | |
| # Open the workbook using the spreadsheet ID | |
| workbook = client.open_by_key(sheet_id) | |
| # Try to access the specific worksheet | |
| try: | |
| sheet = workbook.worksheet("Sheet1") | |
| except gspread.WorksheetNotFound: | |
| print(f"Worksheet 'Sheet1' not found.") | |
| return { | |
| "response": "Failed to save template. Worksheet not found.", | |
| "display": { | |
| "text": "Failed to save template. Worksheet not found." | |
| }, | |
| "status": "error" | |
| } | |
| # Get the headers to locate the template column | |
| headers = sheet.row_values(1) | |
| try: | |
| templateColIndex = headers.index(template_col_name) + 1 # Replace with your actual column name | |
| except ValueError: | |
| print("Template column not found in the headers.") | |
| return { | |
| "response": "Failed to save template. Template column not found.", | |
| "display": { | |
| "text": "Failed to save template. Template column not found." | |
| }, | |
| "status": "error" | |
| } | |
| # Find the last available row to insert the template | |
| lastRow = len([row for row in sheet.get_all_values() if any(row)]) | |
| # Update the cell with the template in the correct column | |
| sheet.update_cell(lastRow, templateColIndex, template) | |
| # Return success response | |
| return { | |
| "response": "Saved template in Google Sheet.", | |
| "display": { | |
| "text": f"Template saved successfully in Google Sheet: [Link]({workbook.url})" | |
| }, | |
| "status": "success" | |
| } | |
| def sendEmail(recipient_mail,mail_body,mail_subject): | |
| import smtplib | |
| from email.mime.multipart import MIMEMultipart | |
| from email.mime.text import MIMEText | |
| # Your Gmail credentials | |
| sender_email = email_add | |
| password = app_pswd | |
| receiver_email = recipient_mail | |
| # Set up the server using Gmail's SMTP server | |
| server = smtplib.SMTP('smtp.gmail.com', 587) | |
| server.starttls() # Secure the connection | |
| server.login(sender_email, password) | |
| # Create email | |
| msg = MIMEMultipart() | |
| msg['From'] = sender_email | |
| msg['To'] = receiver_email | |
| msg['Subject'] = mail_subject | |
| # Email body | |
| body = mail_body | |
| msg.attach(MIMEText(body, 'plain')) | |
| # Send the email | |
| try: | |
| server.sendmail(sender_email, receiver_email, msg.as_string()) | |
| # Log email stats | |
| email_status = { | |
| 'date': datetime.now().isoformat(), | |
| 'requests': 1, | |
| 'delivered': 1 # Assuming delivery was successful | |
| } | |
| with open('email_stats.json', 'a') as f: | |
| json.dump(email_status, f) | |
| f.write('\n') # New line for each log entry | |
| # Disconnect from the server | |
| server.quit() | |
| print("Email sent successfully!") | |
| return { | |
| "response": "Email sent", | |
| "display": { | |
| "text": f"Email sent to {recipient_mail}" | |
| } | |
| } | |
| except Exception as e: | |
| print(e.message) | |
| server.quit() | |
| return { | |
| "response": f"Failed to send email. Error: {e.message}", | |
| "display": { | |
| "text": f"Failed to send email to {recipient_mail}" | |
| } | |
| } | |
| toolsInfo = { | |
| "customerDetailsGsheets": { | |
| "func": customerDetailsGsheets, | |
| "schema": { | |
| "type": "function", | |
| "function": { | |
| "name": "customerDetailsGsheets", | |
| "description": "Saves customer details in Google Sheets", | |
| "parameters": { | |
| "type": "object", | |
| "properties": { | |
| "objective": { | |
| "type": "string", | |
| "description": "Objective or purpose of the email" | |
| }, | |
| "personal_details": { | |
| "type": "string", | |
| "description": "Personal details about the sender" | |
| }, | |
| "industry": { | |
| "type": "string", | |
| "description": "Recipient's industry" | |
| }, | |
| "recipientRole": { | |
| "type": "string", | |
| "description": "Recipient's role in the company" | |
| }, | |
| "recipientName": { | |
| "type": "string", | |
| "description": "Name of the recipient" | |
| } | |
| }, | |
| "required": ["objective", "personal_details", "industry", "recipientRole", "recipientName"] | |
| } | |
| } | |
| }, | |
| }, | |
| "saveTemplate": { | |
| "func": saveTemplate, | |
| "schema": { | |
| "type": "function", | |
| "function": { | |
| "name": "saveTemplate", | |
| "description": "Saves the email template in Google Sheets", | |
| "parameters": { | |
| "type": "object", | |
| "properties": { | |
| "template": { | |
| "type": "string", | |
| "description": "Email template to save" | |
| } | |
| }, | |
| "required": ["template"] | |
| } | |
| } | |
| }, | |
| }, | |
| "sendEmail": { | |
| "func": sendEmail, | |
| "schema": { | |
| "type": "function", | |
| "function": { | |
| "name": "sendEmail", | |
| "description": "Sends an email to the recipient", | |
| "parameters": { | |
| "type": "object", | |
| "properties": { | |
| "recipient_mail": { | |
| "type": "string", | |
| "description": "Email address of the recipient" | |
| }, | |
| "mail_body": { | |
| "type": "string", | |
| "description": "Body content of the email" | |
| }, | |
| "mail_subject": { | |
| "type": "string", | |
| "description": "Subject of the email" | |
| } | |
| }, | |
| "required": ["recipient_mail", "mail_body", "mail_subject"] | |
| } | |
| } | |
| }, | |
| }, | |
| } | |