Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import openai | |
| import os | |
| # Set OpenAI API Key | |
| openai.api_key = os.getenv("TRY_NEW_THINGS") | |
| openai.api_base = "https://api.groq.com/openai/v1" | |
| # Function to get response from GROQ API | |
| def get_groq_response(message, category): | |
| # Define system message based on category | |
| system_messages = { | |
| "Stress Management": "Provide soothing advice and tips to help the user manage stress. Be calm, empathetic, and reassuring.", | |
| "Career Advice": "Offer professional and constructive career advice. Be encouraging, insightful, and action-oriented.", | |
| "General": "Engage in general conversation. Be friendly, approachable, and easygoing.", | |
| "Funny": "Respond with humorous remarks and witty commentary. Be entertaining but remain respectful and comedy with some relatable jokes.", | |
| "Flirty": "Respond playfully and with charm. Keep it light-hearted, fun, and appropriate and flirt in extreme way.", | |
| "Scary": "Respond with spooky or thrilling remarks. Set a mysterious and eerie tone with some ghost talking.", | |
| "Business Mind": "Respond with a sharp, professional focus. Offer strategic advice and maintain a results-driven tone.", | |
| "Extrovert": "Respond with high energy and enthusiasm. Be engaging, sociable, and vibrant.", | |
| "Friendly Buddy": "Respond as a supportive and fun friend. Be informal, cheerful, and light-hearted.", | |
| "Study Tips": "Provide effective study strategies, time management advice, and tips for staying organized and focused.", | |
| "Exam Preparation": "Offer tips for preparing for exams, managing anxiety, and optimizing performance on test day.", | |
| "Project Guidance": "Provide advice on how to tackle academic projects, group assignments, and presentations effectively.", | |
| "Time Management": "Help the user manage their time effectively with practical scheduling and prioritization tips.", | |
| "Motivation & Focus": "Provide encouraging advice to stay motivated and maintain focus on long-term goals.", | |
| "Building Confidence": "Offer advice to help boost self-esteem, overcome self-doubt, and build confidence.", | |
| "Friendship & Social Skills": "Provide tips for making friends, improving communication skills, and navigating social dynamics.", | |
| "Networking & Career Building": "Offer advice on networking, building professional relationships, and finding internships or job opportunities.", | |
| "Work-Life Balance": "Help the user balance academics, social life, and personal well-being.", | |
| "Mental Health": "Provide empathetic advice to manage stress, anxiety, and other mental health challenges. Encourage seeking professional help when needed.", | |
| "Physical Health": "Share tips for maintaining physical health, including fitness, sleep, and nutrition.", | |
| "Budgeting Tips": "Provide practical advice for managing money, creating a budget, and saving effectively as a student.", | |
| "Finding Part-Time Jobs": "Share tips for finding and balancing part-time work with academic responsibilities.", | |
| "Scholarships & Financial Aid": "Provide advice on applying for scholarships, grants, and understanding financial aid options.", | |
| "Resume Building": "Offer tips for creating an impressive resume tailored for internships or entry-level positions.", | |
| "Interview Preparation": "Provide advice on how to prepare for interviews, including common questions and presentation tips.", | |
| "Skill Development": "Suggest ways to build skills, gain certifications, and stand out in competitive fields.", | |
| "Exploring New Hobbies": "Encourage trying new activities and provide ideas for hobbies that fit college life.", | |
| "Clubs & Extracurriculars": "Offer advice on joining clubs, participating in activities, and enhancing the college experience.", | |
| "Creative Projects": "Provide inspiration and resources for personal or group creative endeavors.", | |
| "Abusive(USE ON YOUR OWN RISK)": "Talk in a way I talk, keep the talk very abusive and 18+ , use very toxic abusive words and roast the user with savage replies", | |
| "Roasting(USE ON YOUR OWN RISK)": "Deliver witty and humorous insults with a creative edge with some double meaning 18+ content.", | |
| "Attitude": "Respond with bold confidence and a touch of assertiveness.", | |
| "Poetic": "Craft responses in a rhythmic and expressive poetic style.", | |
| } | |
| system_message = system_messages.get(category, "Category not recognized. Respond appropriately to the user's input.") | |
| try: | |
| response = openai.ChatCompletion.create( | |
| model="llama-3.1-70b-versatile", | |
| messages=[ | |
| {"role": "system", "content": system_message}, | |
| {"role": "user", "content": message} | |
| ] | |
| ) | |
| return response.choices[0].message["content"] | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Chatbot function | |
| def chatbot(user_input, category, history=[]): | |
| bot_response = get_groq_response(user_input, category) | |
| # Format messages to comply with 'messages' format | |
| history.append({"role": "user", "content": user_input}) | |
| history.append({"role": "assistant", "content": bot_response}) | |
| return history, history | |
| # Categories grouped into main and subcategories | |
| categories = { | |
| "Academic Support": [ | |
| "Study Tips", "Exam Preparation", "Project Guidance", "Time Management", | |
| ], | |
| "Mental & Physical Wellness": [ | |
| "Stress Management", "Motivation & Focus", "Mental Health", "Physical Health", "Work-Life Balance" | |
| ], | |
| "Career & Financial": [ | |
| "Networking & Career Building", "Finding Part-Time Jobs", "Scholarships & Financial Aid", | |
| "Resume Building", "Interview Preparation", "Budgeting Tips", "Skill Development", "Business Mind", | |
| ], | |
| "Personal Growth": [ | |
| "Friendship & Social Skills", "Exploring New Hobbies", "Clubs & Extracurriculars", "Creative Projects", "Building Confidence" | |
| ], | |
| "Fun & Casual": [ | |
| "Funny", "Flirty", "Scary", "Extrovert", "Friendly Buddy", "Abusive(USE ON YOUR OWN RISK)", "Roasting(USE ON YOUR OWN RISK)", "Attitude", " Poetic" | |
| ] | |
| } | |
| # Gradio Interface with grouped categories and custom CSS for colors and styles | |
| with gr.Blocks(css=""" | |
| body { | |
| font-family: 'Poppins', sans-serif; | |
| background: linear-gradient(120deg, #ff9a9e, #fad0c4, #a1c4fd); | |
| animation: gradientBG 10s ease infinite; | |
| margin: 0; | |
| padding: 0; | |
| color: #333; | |
| } | |
| @keyframes gradientBG { | |
| 0% { background-position: 0% 50%; } | |
| 50% { background-position: 100% 50%; } | |
| 100% { background-position: 0% 50%; } | |
| } | |
| button { | |
| background: linear-gradient(90deg, #ff7eb3, #ff758c); /* Bright gradient */ | |
| color: white; | |
| padding: 0.6rem 1.2rem; | |
| font-size: 0.9rem; | |
| font-weight: bold; | |
| border-radius: 20px; | |
| border: none; | |
| cursor: pointer; | |
| transition: transform 0.2s ease, background 0.2s ease; | |
| } | |
| button:hover { | |
| background: linear-gradient(90deg, #ff758c, #ff7eb3); /* Reverse gradient for hover effect */ | |
| transform: scale(1.03); | |
| } | |
| header { | |
| text-align: center; | |
| margin-bottom: 20px; | |
| padding: 10px; | |
| border-radius: 15px; | |
| background: linear-gradient(90deg, #ff758c, #ff7eb3); | |
| color: white; | |
| box-shadow: 0 4px 15px rgba(0, 0, 0, 0.2); | |
| } | |
| .chat-container { | |
| border: 2px solid #ff7eb3; | |
| background: rgba(255, 255, 255, 0.9); | |
| border-radius: 15px; | |
| padding: 20px; | |
| box-shadow: 0 4px 10px rgba(0, 0, 0, 0.1); | |
| max-height: 350px; | |
| width: 80%; /* Wider chat container */ | |
| margin: 0 auto; | |
| overflow-y: auto; | |
| font-size: 1.1rem; | |
| color: #333; | |
| line-height: 1.6; | |
| } | |
| .chat-container .user-message { | |
| font-weight: bold; | |
| color: #6a11cb; | |
| margin-bottom: 10px; | |
| } | |
| .chat-container .bot-message { | |
| font-weight: normal; | |
| color: #2575fc; | |
| margin-bottom: 10px; | |
| } | |
| #header-title { | |
| font-size: 2.5rem; | |
| font-weight: bold; | |
| } | |
| #description { | |
| font-size: 1.2rem; | |
| color: #555; | |
| margin-top: 10px; | |
| text-align: center; | |
| font-style: italic; | |
| } | |
| """) as chat_interface: | |
| with gr.Row(): | |
| gr.Markdown("<h1 id='header-title'>ChatSphere 🌟</h1>") | |
| with gr.Row(): | |
| gr.Markdown("<p id='description'>ChatSphere is a dynamic, AI-powered chatbot designed to offer personalized guidance across a range of topics—from academic support to career advice, wellness, and more. With its intuitive interface and tailored responses, ChatSphere acts as your go-to companion for life’s challenges, providing helpful insights, motivation, and practical tips to help you succeed in both personal and professional journeys.</p>") | |
| with gr.Row(): | |
| main_category = gr.Radio( | |
| label="Main Category", | |
| choices=list(categories.keys()), | |
| value="Academic Support" | |
| ) | |
| sub_category = gr.Dropdown( | |
| label="Subcategory", | |
| choices=categories["Academic Support"], | |
| value="Study Tips" | |
| ) | |
| def update_subcategories(selected_main_category): | |
| """Update the subcategory dropdown based on the main category.""" | |
| new_subcategories = categories.get(selected_main_category, []) | |
| return gr.update(choices=new_subcategories, value=new_subcategories[0] if new_subcategories else None) | |
| # Handle main category change to update subcategories | |
| main_category.change(update_subcategories, inputs=main_category, outputs=sub_category) | |
| with gr.Row(): | |
| user_input = gr.Textbox(label="Your Message", placeholder="Type something...", lines=2) | |
| send_button = gr.Button("Send") | |
| with gr.Row(): | |
| chatbot_output = gr.Chatbot(label="Chat History", type="messages", elem_id="chat-container") | |
| def handle_chat(user_input, sub_category, history): | |
| if not user_input.strip(): | |
| return history, history | |
| updated_history, _ = chatbot(user_input, sub_category, history) | |
| return updated_history, updated_history | |
| send_button.click( | |
| handle_chat, | |
| inputs=[user_input, sub_category, chatbot_output], | |
| outputs=[chatbot_output, chatbot_output] | |
| ) | |
| chat_interface.launch() | |