import streamlit as st
import os
from dotenv import load_dotenv
import google.generativeai as genai

# Load environment variables
load_dotenv()
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")

# Validate API Key
if not GOOGLE_API_KEY:
    st.error("Google Gemini API key not found. Please set GOOGLE_API_KEY in your .env file.")
    st.stop()

# Configure Gemini
genai.configure(api_key=GOOGLE_API_KEY)

# Streamlit page setup
st.set_page_config(page_title="NeuraNexus: AI-Powered ML Assistant", page_icon="🧠", layout="wide")

# ==============================
# Custom CSS (include your full CSS here)
# ==============================
st.markdown("""
<style>
body {
    background-color: #0F1C2E;
    color: white;
    font-family: 'Roboto', sans-serif;
}
h1 {
    font-family: 'Orbitron', sans-serif;
    color: #FF6B6B;
    text-align: center;
    font-size: 2.5em;
    text-shadow: 0 0 10px rgba(255, 107, 107, 0.7);
}
.result-container {
    background: rgba(255, 255, 255, 0.05);
    border: 1px solid rgba(74, 144, 226, 0.2);
    border-radius: 15px;
    padding: 20px;
    margin-top: 30px;
    color: white;
    box-shadow: 0 8px 32px 0 rgba(31, 38, 135, 0.2);
    backdrop-filter: blur(10px);
}
.by-theaimart {
    text-align: center;
    font-size: 20px;
    color: #50E3C2;
    margin-top: 20px;
    font-weight: bold;
}
</style>
""", unsafe_allow_html=True)

# ==============================
# App UI
# ==============================
st.markdown('<h1 class="glow">NeuraNexus: AI-Powered ML Assistant</h1>', unsafe_allow_html=True)
st.markdown('<h3 style="text-align:center;">Describe your ML challenge, and let NeuraNexus craft an innovative solution.</h3>', unsafe_allow_html=True)

# Input from user
user_prompt = st.text_area("Your ML Challenge", height=150, placeholder="Enter your machine learning challenge here...")

if 'response' not in st.session_state:
    st.session_state.response = ""

# Generate button
if st.button("SYNTHESIZE"):
    if not user_prompt.strip():
        st.warning("Please describe your ML challenge first.")
    else:
        with st.spinner("NeuraNexus is thinking..."):
            try:
                model = genai.GenerativeModel("gemini-1.5-flash-8b")
                chat = model.start_chat()

                system_prompt = "You are NeuraNexus, an AI specialized in designing innovative and actionable ML strategies. Provide expert-level solutions to the given ML challenge."

                full_prompt = f"{system_prompt}\n\nUser Problem:\n{user_prompt}"

                response = chat.send_message(full_prompt)
                st.session_state.response = response.text
                st.success("Synthesis complete!")

            except Exception as e:
                st.session_state.response = f"❌ Error: {e}"
                st.error("Gemini synthesis failed.")

# ==============================
# Show result
# ==============================
if st.session_state.response:
    st.markdown("### NeuraNexus Synthesis Result")
    st.markdown(f"""
    <div class="result-container">
        {st.session_state.response}
    </div>
    """, unsafe_allow_html=True)

# Footer
st.markdown('<p class="by-theaimart">By Theaimart</p>', unsafe_allow_html=True)

# Run wrapper
def main():
    pass

if __name__ == "__main__":
    main()