app.py

import streamlit as st
from utils import clone_repo
from backend import process_repo, ask_question, generate_summary

st.set_page_config(page_title="AI Codebase Explainer", layout="wide")

# 🎨 UI Styling
st.markdown("""
<style>
.stTextInput input {
    border-radius: 10px;
    padding: 8px;
}
.stButton button {
    background-color: #4CAF50;
    color: white;
    border-radius: 10px;
}
</style>
""", unsafe_allow_html=True)

st.title("💻 AI Codebase Explainer")

# Session state
if "db" not in st.session_state:
    st.session_state.db = None

if "summary" not in st.session_state:
    st.session_state.summary = None

# Input
repo_url = st.text_input("🔗 Enter GitHub Repository URL")

# Analyze Repo
if st.button("🚀 Analyze Repo"):
    if not repo_url.startswith("https://github.com/"):
        st.error("❌ Please enter a valid GitHub URL")
    else:
        with st.spinner("Cloning and analyzing repo..."):
            try:
                path = clone_repo(repo_url)
                db = process_repo(path)
                st.session_state.db = db

                # 🔥 Auto Summary
                summary = generate_summary(db)
                st.session_state.summary = summary

                st.success("✅ Repo analyzed successfully!")
            except Exception as e:
                st.error(f"Error: {e}")

# 📊 Show Summary
if st.session_state.summary:
    st.markdown("## 📊 Project Overview")
    st.write(st.session_state.summary)

# 💬 Chat Section
st.markdown("## 💬 Ask Questions")

query = st.text_input("Ask something about the code")

if st.button("Ask"):
    if st.session_state.db:
        with st.spinner("Thinking..."):
            try:
                answer = ask_question(st.session_state.db, query)
                st.markdown("### 🤖 Answer")
                st.write(answer)
            except Exception as e:
                st.error(f"Error: {e}")
    else:
        st.warning("⚠️ Please analyze a repo first!")