Hugging Face's logo

Spaces:
aheedsajid
/
personality-chat
Running

App Files Community
1
personality-chat
File size: 3,918 Bytes 
76619ab
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4c7d2b8
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
 
import os
import gradio as gr
import google.generativeai as genai
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Retrieve API key from environment variable
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")

# Retrieve system content from environment variable
SYSTEM_CONTENT = os.getenv("SYSTEM_CONTENT")

# Configure Google Gemini API
genai.configure(api_key=GEMINI_API_KEY)

# Create the model
generation_config = {
    "temperature": 0.7,
    "top_p": 0.95,
    "top_k": 64,
    "max_output_tokens": 512,  # Adjust as needed
    "response_mime_type": "text/plain",
}

# Define safety settings for the model
safety_settings = [
    {
        "category": "HARM_CATEGORY_HARASSMENT",
        "threshold": "BLOCK_NONE"
    },
    {
        "category": "HARM_CATEGORY_HATE_SPEECH",
        "threshold": "BLOCK_NONE"
    },
    {
        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
        "threshold": "BLOCK_NONE"
    },
    {
        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
        "threshold": "BLOCK_NONE"
    }
]

# Function to read names and characters from list.txt
def load_characters():
    characters = {}  # Use a dictionary to store names and full descriptions
    with open("list.txt", "r") as f:
        for line in f:
            if "(" in line:
                name = line.split("(")[0].strip()
                description = line.strip()  # Keep the full line with brackets
                characters[name] = description 
            else:
                characters[line.strip()] = line.strip()  # If no brackets, name is the description
    return characters

# Function to generate a response based on user input and chat history
def generate_response(user_input, chat_history, selected_character):
    """Generates a response based on user input, chat history, and selected character."""

    # Get the full character description from the dictionary
    full_character_description = characters.get(selected_character, selected_character)

    # Update system content with the full character description
    updated_system_content = SYSTEM_CONTENT.replace("godfather", full_character_description)

    # Create the generative model
    model = genai.GenerativeModel(
        model_name="gemini-1.5-pro", 
        generation_config=generation_config,
        safety_settings=safety_settings,
        system_instruction=updated_system_content,
    )

    # Add user input to history
    chat_history.append(user_input)

    # Limit history length
    if len(chat_history) > 10:
        chat_history = chat_history[-10:]

    # Start a new chat session
    chat_session = model.start_chat()

    # Send the entire chat history as the first message
    response = chat_session.send_message("\n".join(chat_history))

    # Return response and update chat history
    return response.text, chat_history

# Load characters from list.txt
characters = load_characters()

# Extract character names for the dropdown
character_names = list(characters.keys())

with gr.Blocks() as iface:
    gr.Interface(
        fn=generate_response,
        inputs=[
            gr.Textbox(lines=2, label="Talk to AI", placeholder="Enter your message here..."),
            gr.State([]),  # State input for chat history
            gr.Dropdown(choices=character_names, label="Select Character")
        ],
        outputs=[
            gr.Textbox(label="Response"),
            gr.State([])  # State output to update chat history 
        ],
        title="AI Character Chatbot",
        description="Chat with different AI characters!<br>"
                "Contact me if you want another character/voice<br>"
                "WhatsApp me: +92-332-4399819<br>"
                "Email me: aheedsajid@gmail.com<br>"
                "<b>Donate something to increase GPU power</b><br>"
                "[Click here to Donate](https://nowpayments.io/donation/aheed)")

iface.launch()