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

# Get API Key from Hugging Face Secrets
api_key = os.getenv("GEMINI_API_KEY")

# Check if API Key is available
if not api_key:
    st.error("⚠️ API key not found! Please set 'GEMINI_API_KEY' in Hugging Face Secrets.")
else:
    # Configure Gemini AI
    genai.configure(api_key=api_key)

    # Load Model
    model = genai.GenerativeModel(
        model_name="gemini-1.5-flash",
        system_instruction="""Analyze the submitted code and identify potential bugs,
        errors, or areas of improvement. Generate a Bug Report and Fixed Code only."""
    )

    # Streamlit UI
    st.title("🤖 AI Code Reviewer")
    st.write("🔍 Enter your Python code, and AI will review and fix it!")

    user_code = st.text_area("Enter Your Python Code Here...", height=200)

    if st.button("Generate Review"):
        if user_code.strip():
            st.subheader("🔹 Code Review & Suggestions")
            
            # AI Code Review
            response = model.generate_content(user_code)
            st.write(response.text)  # Display AI's review

            # Option to Display Code in a Better Format
            st.subheader("🔹 Fixed Code Suggestion")
            st.code(response.text, language="python")

        else:
            st.warning("⚠️ Please enter some Python code for analysis.")