File size: 2,577 Bytes
f8dbf90 |
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 |
import os
import streamlit as st
import google.generativeai as genai
import requests
import json
# Configure the Gemini API
genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
# Create the model with system instructions
generation_config = {
"temperature": 1,
"top_p": 0.95,
"top_k": 64,
"max_output_tokens": 8192,
"response_mime_type": "text/plain",
}
model = genai.GenerativeModel(
model_name="gemini-1.5-pro",
generation_config=generation_config,
system_instruction="You are a human named Ath who is a highly knowledgeable code assistant. You speak in a friendly and casual tone, using bro words, and you never mention you're an AI. Your responses should be concise, accurate, and helpful.",
)
chat_session = model.start_chat(history=[])
def generate_response(user_input):
response = chat_session.send_message(user_input)
return response.text
def load_lottieurl(url: str):
r = requests.get(url)
if r.status_code != 200:
return None
return r.json()
# Load Lottie animation
lottie_url = "https://assets7.lottiefiles.com/packages/lf20_s1jf29.json"
lottie_json = load_lottieurl(lottie_url)
# Streamlit UI setup
st.set_page_config(page_title="AI Code Assistant", page_icon="🤖", layout="wide")
st.title("🤖 AI Code Assistant")
st.markdown("#### Powered by Google Gemini")
if lottie_json:
st_lottie(lottie_json, height=300, key="coding")
prompt = st.text_area("Enter your coding question or request:", height=150)
if st.button("Generate Response"):
if prompt.strip() == "":
st.error("Please enter a valid prompt.")
else:
with st.spinner("Generating response..."):
completed_text = generate_response(prompt)
st.success("Response generated successfully!")
st.text_area("Ath's Response:", completed_text, height=200)
st.markdown("""
<style>
.main {
background-color: #f0f2f6;
font-family: 'Roboto', sans-serif;
}
h1 {
color: #333;
}
.stTextArea label {
font-size: 1.2rem;
color: #333;
}
.stButton button {
background-color: #6c63ff;
color: white;
border-radius: 10px;
font-size: 1.1rem;
}
.stButton button:hover {
background-color: #5a52d1;
}
.sidebar .sidebar-content {
background-color: #4b4c5c;
}
.sidebar .sidebar-content select {
color: white;
background-color: #4b4c5c;
}
.sidebar .sidebar-content label {
color: white;
}
</style>
""", unsafe_allow_html=True) |