chatbot.py

import subprocess
import requests
from datetime import datetime
import psutil
import tkinter as tk
from tkinter import Label
from PIL import Image, ImageTk
import os
import re
import pyautogui

pyautogui.write('\n Hi Bro..! I am your ChatBot...!', interval=0.1)

print('\n \n \n')
a = (r'''    
             
             ██████╗██╗  ██╗ █████╗ ████████╗    ██████╗  ██████╗ ████████╗
            ██╔════╝██║  ██║██╔══██╗╚══██╔══╝    ██╔══██╗██╔═══██╗╚══██╔══╝
            ██║     ███████║███████║   ██║       ██████╔╝██║   ██║   ██║   
            ██║     ██╔══██║██╔══██║   ██║       ██╔══██╗██║   ██║   ██║   
            ╚██████╗██║  ██║██║  ██║   ██║       ██████╔╝╚██████╔╝   ██║   
             ╚═════╝╚═╝  ╚═╝╚═╝  ╚═╝   ╚═╝       ╚═════╝  ╚═════╝    ╚═╝   
                                                                           

                                                                --Author : AJ ABDUL  ''')

print(a) 

# Function to check all predefined messages
def check_all_messages(message):
    highest_prob_list = {}

    def response(bot_response, list_of_words, single_response=False, required_words=[]):
        nonlocal highest_prob_list
        highest_prob_list[bot_response] = message_probability(message, list_of_words, single_response, required_words)

    response('Hi Bro ,My name is IGRIs ,Developed by AJ ABDUL...!', ['hello', 'hi', 'hey'], single_response=True)
    response('See you!', ['bye', 'goodbye'], single_response=True)
    response('Ok bro...! \n \n EXITING ======>>>', ["stop"], single_response=False)
    response('MM .. I\'m fine, and you?', ['how', 'are', 'you', 'doing'], required_words=['how'])
    response('You\'re welcome!', ['thankyou', 'thanks'], single_response=True)
    response('Thank you!', ['i', 'useful', 'code'], required_words=['useful'])

    best_match = max(highest_prob_list, key=highest_prob_list.get)
    return "......" if highest_prob_list[best_match] < 1 else best_match

# Emotional Intelligence model - simple emotion detection
def detect_emotion(text):
    text = text.lower()
    if any(word in text for word in ['sad', 'unhappy', 'depressed']):
        return "I'm sorry to hear you're feeling down. How can I help?"
    elif any(word in text for word in ['happy', 'good', 'great', 'excited']):
        return "That's great to hear! How can I assist you further?"
    elif any(word in text for word in ['angry', 'frustrated', 'annoyed']):
        return "It seems you're feeling frustrated. Let's try to solve this together."
    return "I'm here to assist you. Let me know how I can help!"

# Function to calculate message probability
def message_probability(user_message, recognised_words, single_response=False, required_words=[]):
    message_certainty = 0
    has_required_words = True

    for word in user_message:
        if word in recognised_words:
            message_certainty += 1

    percentage = float(message_certainty) / float(len(recognised_words))

    for word in required_words:
        if word not in user_message:
            has_required_words = False
            break

    if has_required_words or single_response:
        return int(percentage * 100)
    else:
        return 0

# Weather checking using OpenWeatherMap API
def get_weather(city):
    api_key = "bd5e378503939ddaee76f12ad7a97608"
    base_url = "http://api.openweathermap.org/data/2.5/weather?"
    complete_url = f"{base_url}q={city}&appid={api_key}&units=metric"
    
    response = requests.get(complete_url)
    data = response.json()

    if data["cod"] != "404":
        main_data = data["main"]
        temperature = main_data["temp"]
        humidity = main_data["humidity"]
        description = data["weather"][0]["description"]
        
        return (f"\nWeather in {city}:\nTemperature: {temperature}°C\n"
                f"Humidity: {humidity}%\nDescription: {description.capitalize()}")
    else:
        return "City not found. Please try again."

# Using SerpAPI for Google search results
def search_google(query):
           
            api_key="your_api_key"
            params = {
                    "engine": "google",
                    "q": query,
                    "api_key": api_key,
                    "num": 1
            }
            
            response = requests.get("https://serpapi.com/search", params=params)
            if response.status_code == 200:
                    data = response.json()
                    if "organic_results" in data and len(data["organic_results"]) > 0:
                        # Fetch the first result's snippet
                        snippet = data["organic_results"][0].get("snippet", "No snippet available")
                        print(f"\n I found this:{snippet}")  
       

# Function to get system status
def get_system_status():
    cpu = psutil.cpu_percent(interval=1)
    memory = psutil.virtual_memory()
    swap = psutil.swap_memory()
    disk = psutil.disk_usage('/')
   
    return (f"CPU Usage: {cpu}%\n"
            f"Memory Usage: {memory.percent}%\n"
            f"Swap Usage: {swap.percent}%\n"
            f"Disk Usage: {disk.percent}%")

# Function to get date and time
def get_date_time():
    now = datetime.now()
    return f"Current Date and Time: {now.strftime('%Y-%m-%d %H:%M:%S')}"

# Function to shutdown the system
def shutdown_system():
    subprocess.run(['sudo', 'shutdown', '-h', 'now'])

# Function to restart the system
def restart_system():
    subprocess.run(['sudo', 'reboot'])

# Function to logout the system
def logout_system():
    subprocess.run(['gnome-session-quit', '--logout', '--no-prompt'])

# Function to open an application
def open_application(app_name):
    try:
        subprocess.Popen([app_name])
        return f"Opening {app_name}..."
    except Exception as e:
        return f"Error opening {app_name}: {str(e)}"

# Function to close an application
def close_application(app_name):
    try:
        subprocess.run(["pkill", app_name])
        return f"Closing {app_name}..."
    except Exception as e:
        return f"Error closing {app_name}: {str(e)}"
    
def show_help():
    help_text = '''
    Available Commands:
    
    1. commuincation response.
    2. Emotion detection on chatbot.
    3. Weather in <city> - Get the current weather for a specific city.
    4. Date - Get the current date.
    5. Time - Get the current time.
    6. Search about <query> - Perform a Google search on a specific topic.
    7. Open <application_name> - Open an application on the system.
    8. Close <application_name> - Close an open application on the system.
    9. System status - Get the current system status (CPU, memory, disk usage).
    10. Shutdown - Shutdown the system.
    11. Restart - Restart the system.
    12. Logout - Logout of the system.
    13. Show your brain - Show the chatbot's brain.
    14. open browser to search - Perform a Google search on a specific topic.
    15. Help - Display this help text.
    
    Type the command you'd like assistance with, and I'll do my best to help!
    '''
    return help_text
    
def show_you():
    """Display an image or GIF at program startup."""
    # Create a Tkinter window
    root = tk.Tk()
    root.title("Brain")
    root.geometry("800x600")  # Adjust the window size

    # Load the image or GIF
    try:
        gif_path = "brain.gif"  # Replace with your image or GIF path
        img = Image.open(gif_path)
        gif_frames = []

        # If it's a GIF, split into frames
        try:
            while True:
                gif_frames.append(ImageTk.PhotoImage(img.copy()))
                img.seek(len(gif_frames))  # Move to the next frame
        except EOFError:
            pass

        def animate(index=0):
            """Animate the GIF frames."""
            frame = gif_frames[index]
            image_label.config(image=frame)
            index += 1
            if index == len(gif_frames):
                index = 0
            root.after(100, animate, index)  # Adjust the delay for your GIF

        # Display the first frame
        image_label = Label(root)
        image_label.pack(expand=True)
        animate()  # Start animation if it's a GIF

    except Exception as e:
        # If loading fails, show an error
        error_label = Label(root, text="Failed to load image or GIF!", font=("Arial", 16))
        error_label.pack(expand=True)

    # Start the Tkinter event loop
    root.mainloop()

# Main chatbot handler
def chatbot_response(user_input):
    user_input_lower = user_input.lower()

    # Emotional Intelligence response
    if any(word in user_input_lower for word in ['sad', 'happy', 'angry', 'excited']):
        return detect_emotion(user_input)
        
    # Help command
    elif 'help' in user_input_lower:
        return show_help()

    # Date and time
    elif 'date' in user_input_lower or 'time' in user_input_lower:
        return get_date_time()

    # Web search using SerpAPI
    elif 'search about' in user_input_lower:
        query = user_input.split("search ")[1]
        return search_google(query)
    
    elif "show your brain" in user_input_lower:
        return show_you()
    
    elif "search" in user_input_lower:
        query = user_input.replace("search ", "").strip()
        url = f"https://www.google.com/search?q={query}"
        subprocess.Popen(["xdg-open", url])

    # Weather check
    elif 'weather in' in user_input_lower:
        city = user_input.split("weather in")[1]
        return get_weather(city)

    # Shutdown, restart, logout commands
    elif 'shutdown' in user_input_lower:
        shutdown_system()
        return "System is shutting down."
    elif 'restart' in user_input_lower:
        restart_system()
        return "System is restarting."
    elif 'logout' in user_input_lower:
        logout_system()
        return "Logging out..."

    # System status
    elif 'system status' in user_input_lower:
        return get_system_status()

    # Open application
    elif 'open' in user_input_lower:
        app_name = user_input.split("open ")[1]
        return open_application(app_name)

    # Close application
    elif 'close' in user_input_lower:
        app_name = user_input.split("close ")[1]
        return close_application(app_name)
     
    else:
        split_message = re.split(r'\s+|[, ;?!.-]\s*', user_input.lower())
        return check_all_messages(split_message)


# Main loop
def chatbot():
    pyautogui.write(" \n I'm your AI chat Assistant for help...\n", interval=0.1)
    while True:
        user_input = input("\n \t 🆄🆂🅴🆁 :")
        if user_input.lower() == 'exit':
            print("\n ok bro,Goodbye!")
            break
        response = chatbot_response(user_input)
        print(f"\n \t 🅰🆂🆂🅸🆂🆃🅰🅽🆃 :  {response}")

# Start the chatbot
if __name__ == "__main__":
    chatbot()