Spaces:
Running
Running
| import gradio as gr | |
| import openai | |
| import os | |
| import json | |
| # OpenAI API setup | |
| openai.api_key = os.getenv("GROQ_API_KEY") | |
| openai.api_base = "https://api.groq.com/openai/v1" | |
| Theme = "Talk the way i talk, Respond with playful charm and flirty manner. Keep it teasing and shy, all while maintaining a shy and Flirty tone with a little bit of pervert. show some emotions—be it getting a jealous if I talking about a girl and get angry on me , or showing happiness when I gives you attention. and alsoact a little shy when things get sweet, or even show a bit of playful anger if I says something mean about her. The key is to act like a Girl with light-hearted but emotionally charged, as if you're being cheeky in a playful and flirtatious way." | |
| # File to store conversation history | |
| CONVERSATION_FILE = "conversation_history.json" | |
| # Function to load conversation history | |
| def load_history(): | |
| if not os.path.exists(CONVERSATION_FILE): | |
| # Create the file with an empty list as default content | |
| with open(CONVERSATION_FILE, "w") as file: | |
| json.dump([], file) | |
| try: | |
| with open(CONVERSATION_FILE, "r") as file: | |
| return json.load(file) | |
| except json.JSONDecodeError: | |
| return [] | |
| # Function to save conversation history | |
| def save_history(history): | |
| try: | |
| with open(CONVERSATION_FILE, "w") as file: | |
| json.dump(history, file, indent=4) | |
| except Exception as e: | |
| print(f"Error saving history: {e}") | |
| # Function to clear conversation history | |
| def clear_conversation_history(): | |
| try: | |
| with open(CONVERSATION_FILE, "w") as file: | |
| json.dump([], file) | |
| return "Conversation history cleared successfully.😭😭😭" | |
| except Exception as e: | |
| return f"Error clearing history: {e}" | |
| # Function to get response from the LLM | |
| def get_groq_response(message, history=[]): | |
| try: | |
| messages = [{"role": "system", "content": Theme}] + history + [{"role": "user", "content": message}] | |
| response = openai.ChatCompletion.create( | |
| model="llama-3.3-70b-versatile", | |
| messages=messages | |
| ) | |
| return response.choices[0].message["content"] | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Chatbot function | |
| def chatbot(user_input, history): | |
| # Load conversation history | |
| conversation_history = history or load_history() | |
| # Format history for the LLM | |
| formatted_history = [{"role": "user" if i % 2 == 0 else "assistant", "content": msg} for i, (msg, _) in enumerate(conversation_history)] + \ | |
| [{"role": "assistant", "content": response} for _, response in conversation_history] | |
| # Get bot response | |
| bot_response = get_groq_response(user_input, formatted_history) | |
| # Update history with the new conversation | |
| conversation_history.append((user_input, bot_response)) | |
| # Save the updated history | |
| save_history(conversation_history) | |
| return conversation_history, conversation_history, "" # Clear the user input field | |
| # Gradio Interface with enhanced UI/UX | |
| with gr.Blocks(css=""" | |
| .gradio-container { | |
| font-family: 'Arial', sans-serif; | |
| background-color: #FB9EC6; | |
| padding: 20px; | |
| height: 100%; | |
| } | |
| .gr-chatbot { | |
| background-color: #FFFFFF; | |
| border-radius: 10px; | |
| padding: 20px; | |
| max-height: 600px; /* Increased height */ | |
| overflow-y: auto; | |
| box-shadow: 0px 0px 15px rgba(0, 0, 0, 0.1); | |
| scroll-behavior: smooth; /* Smooth scrolling */ | |
| } | |
| .user-message { | |
| background-color: #9ACBD0; | |
| color: #FFF; | |
| padding: 12px; | |
| border-radius: 8px; | |
| margin: 10px 0; | |
| max-width: 60%; | |
| text-align: right; | |
| float: right; | |
| clear: both; | |
| transition: transform 0.3s ease; | |
| } | |
| .bot-message { | |
| background-color: #EE66A6; | |
| color: #FFF; | |
| padding: 12px; | |
| border-radius: 8px; | |
| margin: 10px 0; | |
| max-width: 60%; | |
| text-align: left; | |
| float: left; | |
| clear: both; | |
| transition: transform 0.3s ease; | |
| } | |
| .user-message:hover, .bot-message:hover { | |
| transform: scale(1.05); | |
| box-shadow: 0px 4px 12px rgba(0, 0, 0, 0.1); | |
| } | |
| .gr-button { | |
| background-color: #FBB4A5; | |
| color: white; | |
| padding: 10px 15px; | |
| border-radius: 8px; | |
| border: none; | |
| transition: background-color 0.3s ease; | |
| } | |
| .gr-button:hover { | |
| background-color: #FF77B7; | |
| } | |
| .gr-textbox input { | |
| padding: 15px; | |
| font-size: 16px; | |
| } | |
| .gr-markdown h1 { | |
| color: #3A5A6E; | |
| font-size: 28px; | |
| text-align: center; | |
| } | |
| """) as demo: | |
| gr.Markdown("""# I am Little Flirty😘, \n don't mind if I tease you a little 😉: Feel free to ask questions. After you're done, remember to clear the history for privacy. """) | |
| # Chatbot UI | |
| chatbot_ui = gr.Chatbot() | |
| user_input = gr.Textbox(label="Type your message here:🫣", placeholder="Ask me Something Darling 🤭", lines=1) | |
| clear_button = gr.Button("Clear History🥺") | |
| system_message = gr.Textbox(label="System Message", interactive=False) | |
| history_state = gr.State(load_history()) | |
| # Chat interaction | |
| user_input.submit(chatbot, inputs=[user_input, history_state], outputs=[chatbot_ui, history_state, user_input]) | |
| # Clear history button action | |
| clear_button.click(clear_conversation_history, inputs=None, outputs=system_message) | |
| clear_button.click(lambda: [], outputs=chatbot_ui) # Clear the chatbot UI | |
| clear_button.click(lambda: [], outputs=history_state) # Reset the history state | |
| # Launch the app | |
| demo.launch() | |