Spaces:
Sleeping
Sleeping
File size: 4,689 Bytes
4b9bbd9 d9c32c5 4b9bbd9 23d9f29 35da143 6603185 670e0d1 6603185 4b9bbd9 5558c42 4b9bbd9 d9c32c5 5558c42 d9c32c5 5558c42 e09c2ec d9c32c5 7aa26ef d9c32c5 23d9f29 d9c32c5 4bcc560 35da143 cfb8872 8312a74 cfb8872 ae27e7f 35da143 8312a74 4b9bbd9 d9c32c5 4b9bbd9 a494892 |
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 |
import os
import gradio as gr
import google.generativeai as genai
from dotenv import load_dotenv
import time
DESCRIPTIONx = """## GEMINI::GEN ♊
"""
css = '''
.gradio-container{max-width: 680px !important}
h1{text-align:center}
footer {
visibility: hidden
}
'''
load_dotenv()
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
genai.configure(api_key=GEMINI_API_KEY)
generation_config = {
"temperature": 0.7,
"top_p": 0.95,
"top_k": 64,
"max_output_tokens": 512,
"response_mime_type": "text/plain",
}
mood_prompts = {
"Fun": "Respond in a light-hearted, playful manner.",
"Serious": "Respond in a thoughtful, serious tone.",
"Professional": "Respond in a formal, professional manner.",
"Upset": "Respond in a slightly irritated, upset tone.",
"Empathetic": "Respond in a warm and understanding tone.",
"Optimistic": "Respond in a positive, hopeful manner.",
"Sarcastic": "Respond with a hint of sarcasm.",
"Motivational": "Respond with encouragement and motivation.",
"Curious": "Respond with a sense of wonder and curiosity.",
"Humorous": "Respond with a touch of humor.",
"Cautious": "Respond with careful consideration and caution.",
"Assertive": "Respond with confidence and assertiveness.",
"Friendly": "Respond in a warm and friendly manner.",
"Romantic": "Respond with affection and romance.",
"Nostalgic": "Respond with a sense of longing for the past.",
"Grateful": "Respond with gratitude and appreciation.",
"Inspirational": "Respond with inspiration and positivity.",
"Casual": "Respond in a relaxed and informal tone.",
"Formal": "Respond with a high level of formality.",
"Pessimistic": "Respond with a focus on potential negatives.",
"Excited": "Respond with enthusiasm and excitement.",
"Melancholic": "Respond with a sense of sadness or longing.",
"Confident": "Respond with self-assurance and confidence.",
"Suspicious": "Respond with caution and doubt.",
"Reflective": "Respond with deep thought and introspection.",
"Joyful": "Respond with happiness and joy.",
"Mysterious": "Respond with an air of mystery and intrigue.",
"Aggressive": "Respond with force and intensity.",
"Calm": "Respond with a sense of peace and tranquility.",
"Gloomy": "Respond with a sense of sadness or pessimism.",
"Encouraging": "Respond with words of support and encouragement.",
"Sympathetic": "Respond with understanding and compassion.",
"Disappointed": "Respond with a tone of disappointment.",
"Proud": "Respond with a sense of pride and accomplishment.",
"Playful": "Respond in a fun and playful manner.",
"Inquisitive": "Respond with curiosity and interest.",
"Supportive": "Respond with reassurance and support.",
"Reluctant": "Respond with hesitation and reluctance.",
"Confused": "Respond with uncertainty and confusion.",
"Energetic": "Respond with high energy and enthusiasm.",
"Relaxed": "Respond with a calm and laid-back tone.",
"Grumpy": "Respond with a touch of irritation.",
"Hopeful": "Respond with a sense of hope and optimism.",
"Indifferent": "Respond with a lack of strong emotion.",
"Surprised": "Respond with shock and astonishment.",
"Tense": "Respond with a sense of urgency or anxiety.",
"Enthusiastic": "Respond with eagerness and excitement.",
"Worried": "Respond with concern and apprehension."
}
def generate_response(user_input, chat_history, mood):
updated_system_content = f"{mood_prompts[mood]}"
model = genai.GenerativeModel(
model_name="gemini-1.5-pro",
generation_config=generation_config,
system_instruction=updated_system_content,
)
chat_history.append(user_input)
chat_history = chat_history[-10:]
try:
chat_session = model.start_chat()
response = chat_session.send_message("\n".join(chat_history))
return response.text, chat_history
except Exception as e:
return f"Error: {str(e)}", chat_history
with gr.Blocks(css=css, theme="allenai/gradio-theme") as iface:
gr.Markdown(DESCRIPTIONx)
chat_input = gr.Textbox(lines=4, label="Chatbot", placeholder="Enter your message here...")
chat_history_state = gr.State([])
response_output = gr.Textbox(label="Response", lines=4)
generate_button = gr.Button("Submit")
mood_selector = gr.Radio(choices=list(mood_prompts.keys()), value="Professional", label="Select Mood")
generate_button.click(
fn=generate_response,
inputs=[chat_input, chat_history_state, mood_selector],
outputs=[response_output, chat_history_state]
)
iface.launch() |