import streamlit as st
import tempfile
import os
from utils.apk_analyzer import ApkAnalyzer
from utils.report_generator import ReportGenerator

st.set_page_config(page_title="APK Analyzer", page_icon="🔍", layout="wide")

# Sidebar
with st.sidebar:
    st.title("APK Security Analyzer")
    st.write("Upload an APK file for security analysis")
    uploaded_file = st.file_uploader("Choose an APK file", type=["apk"])

    if uploaded_file:
        st.success("File uploaded successfully!")
        file_details = {
            "Filename": uploaded_file.name,
            "File size": f"{uploaded_file.size / 1024:.2f} KB",
        }
        st.write("File Details:")
        for key, value in file_details.items():
            st.write(f"- {key}: {value}")

if uploaded_file:
    with tempfile.NamedTemporaryFile(delete=False, suffix=".apk") as tmp_file:
        tmp_file.write(uploaded_file.getvalue())
        tmp_file_path = tmp_file.name

    with st.spinner("Analyzing APK..."):
        try:
            analyzer = ApkAnalyzer(tmp_file_path)

            # Create two columns for main content
            col1, col2 = st.columns(2)

            with col1:
                with st.container():
                    st.header("Basic Information")
                    basic_info = analyzer.basic_info()
                    st.json(basic_info)

            with col2:
                with st.container():
                    st.header("Security Issues")
                    security_issues = analyzer.security_analysis()
                    for issue in security_issues:
                        st.warning(issue, icon="⚠️")

            # Bottom section for AI Analysis
            st.markdown("---")
            with st.container():
                st.header("AI Analysis Report")

                # Create tabs for different views
                tab1, tab2 = st.tabs(["Report", "Summary"])

                with tab1:
                    report_gen = ReportGenerator()
                    report = report_gen.generate_report(basic_info, security_issues)
                    st.write(report)

                    # Download report button with improved styling
                    st.download_button(
                        label="📥 Download Full Report",
                        data=report,
                        file_name="security_report.txt",
                        mime="text/plain",
                    )

                with tab2:
                    st.subheader("Analysis Summary")
                    
                    # Get existing summary from session state or initialize empty
                    if 'user_summary' not in st.session_state:
                        st.session_state.user_summary = ""
                    
                    # Text area for user input
                    user_summary = st.text_area(
                        "Write your analysis summary here:",
                        value=st.session_state.user_summary,
                        height=300,
                        placeholder="Enter your observations, findings, and recommendations..."
                    )
                    
                    # Save button for the summary
                    if st.button("Save Summary"):
                        st.session_state.user_summary = user_summary
                        st.success("Summary saved successfully!")
                    
                    # Option to download the summary
                    if user_summary:
                        st.download_button(
                            label="📥 Download Summary",
                            data=user_summary,
                            file_name="apk_analysis_summary.txt",
                            mime="text/plain"
                        )

        except Exception as e:
            st.error(f"Error analyzing APK: {str(e)}")
        finally:
            os.unlink(tmp_file_path)
else:
    # Display welcome message when no file is uploaded
    st.markdown(
        """
        <div style='text-align: center; padding: 50px;'>
            <h2>Welcome to APK Security Analyzer</h2>
            <p>Upload an APK file using the sidebar to begin analysis</p>
        </div>
    """,
        unsafe_allow_html=True,
    )