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

# Configure API
genai.configure(api_key="AIzaSyBiAW2GQLid0HGe9Vs_ReKwkwsSVNegNzs")

# Define system prompt
sys_prompt = os.getenv("USER_PROMPT")

# Initialize the model
model = genai.GenerativeModel(model_name="models/gemini-2.5-flash", system_instruction=sys_prompt)

# Streamlit UI Configuration
st.set_page_config(page_title="Python Code Reviewer 🤖", layout="centered")

# Title & Introduction
st.title("🐍 Python Code Reviewer 🤖")
st.markdown("### Welcome! I can help you debug and improve your Python code. Just paste your code below! 🚀")

# Input Text Area
user_query = st.text_area("💬 Enter your Python-related code snippet or query:", height=200)

# Generate Response Button
if st.button("🚀 Review Code"):
    if user_query.strip():
        with st.spinner("Analyzing your code... 🧐"):
            response = model.generate_content(user_query)
        
        # Display Response
        st.markdown("### 🤖 Response:")
        st.success(response.text)
    else:
        st.warning("⚠️ Please enter a Python-related code snippet!")

# Footer
st.markdown("---")
st.markdown("🛠️ **Built with Streamlit & Google Gemini AI** | ❤️ _Happy Coding!_ 🐍")