Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import openai | |
| import google.generativeai as genai | |
| # -------------------------- | |
| # 1. Page Configuration | |
| # -------------------------- | |
| st.set_page_config(page_title="AI-Powered Code Reviewer", layout="wide") | |
| st.title("AI-Powered Python Code Reviewer") | |
| # -------------------------- | |
| # 2. Sidebar API Selection | |
| # -------------------------- | |
| st.sidebar.header("⚙️ Settings") | |
| api_choice = st.sidebar.radio("Choose API:", ["OpenAI", "Google Gemini"]) | |
| if api_choice == "OpenAI": | |
| api_key = st.sidebar.text_input("Enter OpenAI API Key", type="password") | |
| if api_key: | |
| openai.api_key = api_key | |
| else: | |
| api_key = st.sidebar.text_input("Enter Google API Key", type="password") | |
| if api_key: | |
| genai.configure(api_key=api_key) | |
| # -------------------------- | |
| # 3. User Code Input | |
| # -------------------------- | |
| st.subheader("✍️ Enter Your Python Code") | |
| user_code = st.text_area("Paste Python code here:", height=300) | |
| # -------------------------- | |
| # 4. Code Review Function | |
| # -------------------------- | |
| def review_code_with_openai(code): | |
| response = openai.ChatCompletion.create( | |
| model="gpt-4o-mini", # You can change model (gpt-4o, gpt-4o-mini) | |
| messages=[ | |
| {"role": "system", "content": "You are a Python code reviewer. Identify bugs and suggest fixes."}, | |
| {"role": "user", "content": f"Review this Python code:\n{code}"} ], temperature=0.3, ) | |
| return response["choices"][0]["message"]["content"] | |
| def review_code_with_gemini(code): | |
| model = genai.GenerativeModel("gemini-2.5-pro") | |
| response = model.generate_content( f"Review this Python code, detect bugs, and suggest fixes with corrected code:\n{code}" ) | |
| return response.text | |
| # -------------------------- | |
| # 5. Run Code Review | |
| # -------------------------- | |
| if st.button("🔍 Review Code"): | |
| if not user_code.strip(): | |
| st.warning("⚠️ Please enter some Python code first.") | |
| elif not api_key: | |
| st.error("❌ Please enter your API key in the sidebar.") | |
| else: | |
| with st.spinner("Analyzing your code..."): | |
| try: | |
| if api_choice == "OpenAI": | |
| feedback = review_code_with_openai(user_code) | |
| else: | |
| feedback = review_code_with_gemini(user_code) | |
| st.subheader("📢 Review Feedback & Fixes") | |
| st.write(feedback) | |
| except Exception as e: | |
| st.error(f"Error: {str(e)}") |