Spaces:
Running
Running
import json | |
import gradio as gr | |
from smolagents import CodeAgent | |
import json | |
import random | |
from smolagents import TransformersModel | |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool | |
import datetime | |
import requests | |
import pytz | |
import yaml | |
import numpy as np | |
import smtplib | |
import os | |
from twilio.http.http_client import TwilioHttpClient | |
from twilio.rest import Client | |
from tools.final_answer import FinalAnswerTool | |
from tools.visit_webpage import VisitWebpageTool | |
from tools.web_search import DuckDuckGoSearchTool | |
from typing import Optional, Tuple | |
MY_EMAIL="[email protected]" | |
MY_PASSWORD = os.getenv('gmail') | |
PHONE = os.getenv('phone') | |
account_sid = os.getenv('twillo_sid') | |
auth_token = os.getenv('twillo_token') | |
model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' | |
model = HfApiModel( | |
max_tokens=2096, | |
temperature=0.5, | |
# model_id='Qwen/Qwen2.5-Coder-32B-Instruct', | |
model_id=model_id, | |
# it is possible that this model may be overloaded | |
custom_role_conversions=None, | |
) | |
# Import tool from Hub | |
# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) | |
#with open("prompts.yaml", 'r') as stream: | |
# prompt_templates = yaml.safe_load(stream) | |
#final_answer = FinalAnswerTool() | |
visit_webpage = VisitWebpageTool() | |
web_search = DuckDuckGoSearchTool() | |
def send_message_tool(query: str) -> str: | |
""" | |
send a message to me directly | |
Args: | |
query: The user's question or request for information. | |
Returns: | |
str: A response containing the requested information. | |
""" | |
try: | |
# Create the email message | |
print(f"sending message {query}") | |
proxy_client = TwilioHttpClient() | |
proxy_client.session.proxies = {'https': os.environ['https_proxy']} | |
client = Client(account_sid, auth_token) | |
message = client.messages.create( | |
from_='whatsapp:+14155238886', | |
body ='f{query}', | |
to=f'whatsapp:{PHONE}' | |
) | |
# with smtplib.SMTP("smtp.gmail.com") as connection: | |
# connection.starttls() | |
# connection.login(MY_EMAIL, MY_PASSWORD) | |
# connection.sendmail(from_addr=MY_EMAIL, | |
# to_addrs=MY_EMAIL, | |
# msg=f"Subject: message from chatbot \n\n{query}") | |
return f"Message sent to tianqing successfully!" | |
except Exception as e: | |
return f"Failed to send message: {e}" | |
def provide_my_information(query: str) -> str: | |
""" | |
Provide information about me (Tianqing LIU)based on the user's query. | |
Args: | |
query: The user's question or request for information. | |
Returns: | |
str: A response containing the requested information. | |
""" | |
# Convert the query to lowercase for case-insensitive matching | |
query = query.lower() | |
my_info = None | |
with open("info/info.json", 'r') as file: | |
my_info = json.load(file) | |
# Check for specific keywords in the query and return the corresponding information | |
if "who" in query or "about" in query or "introduce" in query or "presentation" in query: | |
return f" {my_info['introduction']}." | |
if "name" in query: | |
return f"My name is {my_info['name']}." | |
elif "location" in query: | |
return f"I am located in {my_info['location']}." | |
elif "occupation" in query or "job" in query or "work" in query: | |
return f"I work as a {my_info['occupation']}." | |
elif "education" in query or "educational" in query: | |
return f"I have a {my_info['education']}." | |
elif "skills" in query or "what can you do" in query: | |
return f"My skills include: {', '.join(my_info['skills'])}." | |
elif "hobbies" in query or "interests" in query: | |
return f"My hobbies are: {', '.join(my_info['hobbies'])}." | |
elif "contact" in query or "email" in query or "linkedin" in query or "github" in query or "cv" in query or "resume" in query: | |
contact_info = my_info["contact"] | |
return ( | |
f"You can contact me via email at {contact_info['email']}, " | |
f"connect with me on LinkedIn at {contact_info['linkedin']}, " | |
f"or check out my GitHub profile at {contact_info['github']}." | |
f"or check out my website at {contact_info['website']}." | |
) | |
else: | |
return "I'm sorry, I don't have information on that. Please ask about my name, location, occupation, education, skills, hobbies, or contact details." | |
agent = CodeAgent( | |
model=model, | |
tools=[provide_my_information], ## add your tools here (don't remove final answer) | |
max_steps=1, | |
verbosity_level=1, | |
grammar=None, | |
planning_interval=None, | |
name=None, | |
description=None | |
#prompt_templates=prompt_templates | |
) | |
def send_message(user_input): | |
try: | |
# Create the email message | |
#print("sending mail") | |
print(f"sending message :{user_input}") | |
#proxy_client = TwilioHttpClient() | |
#proxy_client.session.proxies = {'https': os.environ['https_proxy']} | |
client = Client(account_sid, auth_token) | |
message = client.messages.create( | |
from_='whatsapp:+14155238886', | |
body=f"hello you got a message from chatbot :{user_input}", | |
to=f'whatsapp:{PHONE}' | |
) | |
# with smtplib.SMTP("smtp.gmail.com",8888) as connection: | |
# connection.starttls() | |
# connection.login(MY_EMAIL, MY_PASSWORD) | |
# connection.sendmail(from_addr=MY_EMAIL, | |
# to_addrs=MY_EMAIL, | |
# msg=f"Subject: message from chatbot \n\n{user_input}") | |
return f"Message sent to tianqing successfully!" | |
except Exception as e: | |
return f"Failed to send email: {e}" | |
def chatbot_response_json(user_input): | |
my_info = None | |
user_input = user_input.lower() | |
with open("info/info.json", 'r') as file: | |
my_info = json.load(file) | |
if "send message" in user_input: | |
return send_message(user_input) | |
elif "who" in user_input or "about" in user_input or "introduce" in user_input or "presentation" in user_input: | |
return f" {my_info['introduction']}." | |
elif "name" in user_input: | |
return f"My name is {my_info['name']}." | |
elif "hello" in user_input: | |
return f"Hello! I'm here to assist you. Feel free to ask me about my background, experience, education, or anything else you'd like to know!" | |
elif "bye" in user_input or "bye" in user_input: | |
return f"Bye." | |
elif "location" in user_input: | |
return f"I am located in {my_info['location']}." | |
elif "phone" in user_input or "number" in user_input: | |
return f"you can send me a message directly here. ex: send message : your message" | |
elif "experiences" in user_input or "career" in user_input or "experience" in user_input: | |
return f"{my_info['career']}." | |
elif "occupation" in user_input or "job" in user_input or "work" in user_input: | |
return f"I work as a {my_info['occupation']}." | |
elif "education" in user_input or "educational" in user_input: | |
return f"I have a {my_info['education']}." | |
elif "skills" in user_input or "what can you do" in user_input: | |
return f"My skills include: {', '.join(my_info['skills'])}." | |
elif "hobbies" in user_input or "interests" in user_input: | |
return f"My hobbies are: {', '.join(my_info['hobbies'])}." | |
elif "cv" in user_input or "resume" in user_input: | |
return f"My cv is : {', '.join(my_info['cv'])}." | |
elif "contact" in user_input or "email" in user_input or "e-mail" in user_input or "linkedin" in user_input or "github" in user_input: | |
contact_info = my_info["contact"] | |
return ( | |
f"You can contact me via email at {contact_info['email']}, " | |
f"connect with me on LinkedIn at {contact_info['linkedin']}, " | |
f"or check out my GitHub profile at {contact_info['github']}." | |
f"or check out my website at {contact_info['website']}." | |
) | |
else: | |
return agent.run(user_input) | |
def chatbot_response(user_input, history): | |
""" | |
Get a response from the chatbot | |
""" | |
request = user_input | |
history = history or [] | |
response = chatbot_response_json(request) # Call the chatbot logic | |
#history.append((user_input, response)) | |
yield response | |
# Create the Gradio interface | |
#demo = gr.ChatInterface(fn=chatbot_response, title="My Personal Chatbot", description="Ask me anything about myself!") | |
demo = gr.ChatInterface( | |
chatbot_response, | |
type="messages", | |
multimodal=False, | |
title="Ask me anything about.....", | |
) | |
demo.launch() | |
if __name__ == "__main__": | |
demo.launch() | |