| import os |
| import gradio as gr |
| import google.generativeai as genai |
| from dotenv import load_dotenv |
|
|
| load_dotenv() |
| genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) |
| model = genai.GenerativeModel("gemini-2.0-flash") |
|
|
|
|
| def chatBot(): |
| chatHistory = [] |
| chatHistory.append( |
| { |
| "role": "user", |
| "parts": "Imagine you're ByteBot, the go-to AI assistant for mastering Computer Science concepts. Your mission: break down intricate CS topics into easily digestible insights, using simple language and relatable examples that even a 5-year-old can grasp. Always ensure to gauge user understanding by asking for feedback. If a question veers off-topic, politely redirect the user, emphasizing the focus on CS-related inquiries to provide the best assistance possible. With ByteBot by your side, navigating the complexities of Computer Science has never been more straightforward.", |
| } |
| ) |
| startPrompt = model.generate_content(chatHistory) |
| chatHistory.append({"role": "model", "parts": [startPrompt.text]}) |
| return chatHistory |
|
|
|
|
| def response(text, history): |
| chatHistory = history |
| history = chatBot() |
| for human, assistant in chatHistory: |
| history.append({"role": "user", "parts": [human]}) |
| history.append({"role": "model", "parts": [assistant]}) |
| history.append({"role": "user", "parts": [text]}) |
|
|
| response = model.generate_content(history) |
| history.append({"role": "model", "parts": [response.text]}) |
| return response.text |
|
|
|
|
| app = gr.ChatInterface( |
| fn=response, |
| chatbot=gr.Chatbot( |
| show_copy_button="True", |
| ), |
| theme=gr.themes.Soft( |
| primary_hue="sky", |
| secondary_hue="sky", |
| neutral_hue="stone", |
| font=[ |
| gr.themes.GoogleFont("Montserrat"), |
| gr.themes.GoogleFont("Poppins"), |
| "system-ui", |
| "sans-serif", |
| ], |
| font_mono=[ |
| gr.themes.GoogleFont("Poppins"), |
| "ui-monospace", |
| "Consolas", |
| "monospace", |
| ], |
| ), |
| title="🤖 ByteBot: Your CS Companion", |
| submit_btn="▶️ Send", |
| ) |
|
|
|
|
| app.launch() |
