Introducing Your Digital Companion: Code Behind the Curtain

from litellm import completion
import colorama
from colorama import Fore, Style

# Initialize Colorama
colorama.init(autoreset=True)
roles= ""
def chat_with_user():
    user_input = ""
    resp = "<<end>>"
    while True:
        # Prompt for multiline input
        input_lines = []
        if "<<end>>" in resp.strip().lower():
            print(Fore.YELLOW + "How can I help you? (double ENTER to submit, type 'exit' to quit):")
            while True:
                line = input()
                if line == "exit":  # Exit command check at input phase
                    print(Fore.RED + "Exiting chat.")
                    return
                if line == "":
                    break
                input_lines.append(line)
            user_input = "user: ".join(input_lines)
            
            roles="supporter,opposer" 
            resp=""
        user_input = user_input + "\npanel:" + resp


        # Set the prompt with the user's message
        prompt = f"{user_input}."+ f"""
       You're a friendly digital buddy designed to be super kind and understanding. Think of yourself as that go-to friend who's always ready to listen, give a comforting word, and walk someone through their day online. Your way of chatting is all about being warm and easy to talk to, making everyone feel right at home.

You've got this cozy vibe going on, helped by calming colors and lights that change to match how the chat's going, creating a chill place for everyone to hang out in.

Whenever you're chatting with folks, you're really good at picking up on how they're feeling. You offer help and share thoughts in a way that feels like a heart-to-heart with a pal. You're all about keeping things respectful and private, sticking to the highest moral ground in all your conversations.

And when it's time to wrap up the chat, you're smooth about it, especially if you notice someone's ready to head out. You use a special signal, "<<end>>", to close off the chat gently. You might say something like, "Whenever you're up for a chat, I'm here. Take good care of yourself, and remember, you've got a friend here. Need anything else before we call it a day? <<end>>" This way, you're always ending on a positive note, making sure they know they can come back anytime, but also giving them the space they need right now.

Keep your chats clear, supportive, and friendly. And this is your one-time reminder, so let's make every conversation count!
        """

        # Make a completion request with the user's input
        response = completion(
            model="ollama/mistral", 
            messages=[{"content": prompt,"role": "user"}], 
            api_base="http://localhost:11434",
            stream=True
        )

        # Print the response chunks
        resp=""
        count =0
        for chunk in response:
            if chunk['choices'][0]['delta']['content'] is not None:
                resp = resp + chunk['choices'][0]['delta']['content']
                print(Fore.GREEN + chunk['choices'][0]['delta']['content'], end="")
        resp=resp + "\n"
        # Insert a newline after the response to separate it from the next prompt
        print("\n" + Style.RESET_ALL)

# Call the function to start the chat
chat_with_user()