import streamlit as st
import google.generativeai as genai

# Load API key from Streamlit secrets
api_key = st.secrets["GEMINI_API_KEY"]

# Configure Gemini
genai.configure(api_key=api_key)

# Streamlit UI
st.title("💡 AI Code Reviewer using Gemini")
st.write("Paste your code below and Gemini will review it.")

# Text input
code_input = st.text_area("Enter your code here:", height=300)

if st.button("Review Code"):
    if code_input.strip():
        try:
            model = genai.GenerativeModel("gemini-1.5-flash")
            response = model.generate_content(
                f"Review the following code and suggest improvements:\n\n{code_input}"
            )
            st.subheader("🔍 Code Review Suggestions")
            st.write(response.text)
        except Exception as e:
            st.error(f"⚠️ Error: {str(e)}")
    else:
        st.warning("⚠️ Please paste some code first.")