import streamlit as st
import google.generativeai as genai
import requests

# 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 contain code only, without explanations.",
)
chat_session = model.start_chat(history=[])

def generate_response(user_input):
    response = chat_session.send_message(user_input)
    return response.text

# Streamlit UI setup
st.set_page_config(page_title="AI Code Assistant", page_icon="🤖", layout="wide")

st.markdown("""
<style>
    @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap');
    
    body {
        font-family: 'Roboto', sans-serif;
        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
        color: #ffffff;
    }
    .stApp {
        max-width: 1000px;
        margin: 0 auto;
        padding: 2rem;
        background: rgba(255, 255, 255, 0.1);
        border-radius: 20px;
        backdrop-filter: blur(10px);
    }
    h1 {
        font-size: 3rem;
        font-weight: 700;
        text-align: center;
        margin-bottom: 2rem;
        text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
    }
    .stTextArea label {
        font-size: 1.2rem;
        font-weight: 500;
        color: #ffffff;
    }
    .stTextArea textarea {
        background: rgba(255, 255, 255, 0.2);
        border: none;
        border-radius: 10px;
        color: #ffffff;
        font-size: 1rem;
        padding: 1rem;
    }
    .stButton button {
        background: linear-gradient(45deg, #FF6B6B, #4ECDC4);
        color: white;
        border: none;
        border-radius: 30px;
        font-size: 1.1rem;
        font-weight: 600;
        padding: 0.7rem 2rem;
        transition: all 0.3s ease;
        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
    }
    .stButton button:hover {
        transform: translateY(-2px);
        box-shadow: 0 6px 8px rgba(0,0,0,0.15);
    }
    .stCodeBlock {
        background-color: rgba(0, 0, 0, 0.6);
        color: #f8f8f2;
        font-size: 0.9rem;
        border-radius: 10px;
        padding: 1rem;
        margin-top: 1rem;
        overflow-x: auto;
    }
    .stAlert {
        background-color: rgba(255, 255, 255, 0.2);
        color: #ffffff;
        border-radius: 10px;
        border: none;
    }
    .stSpinner {
        color: #ffffff;
    }
</style>
""", unsafe_allow_html=True)

st.title("🤖 AI Code Assistant")
st.markdown("<h3 style='text-align: center; color: #ffffff;'>Powered by Google Gemini</h3>", unsafe_allow_html=True)

prompt = st.text_area("Enter your coding question or request:", height=150)

if st.button("Generate Code"):
    if prompt.strip() == "":
        st.error("Please enter a valid prompt.")
    else:
        with st.spinner("Generating code..."):
            completed_text = generate_response(prompt)
            st.success("Code generated successfully!")
            
            # Split the response into separate code blocks based on a custom delimiter
            code_blocks = completed_text.split("\n\n")
            for block in code_blocks:
                st.code(block, language="python")

st.markdown("""
<div style='text-align: center; margin-top: 2rem; color: rgba(255,255,255,0.7);'>
    Created with ❤️ by Your AGI Code Assistant
</div>
""", unsafe_allow_html=True)