# sdlc_feedback.py
import streamlit as st

class SDLCUI:
    def render(self):
        if "state" not in st.session_state:
            st.session_state.state = {"current_step": "user_input"}
        state = st.session_state.state

        st.markdown(f"**Current Step:** {state['current_step'].replace('_', ' ').title()}")

        step_method = getattr(self, f"render_{state['current_step']}", None)
        if step_method:
            step_method(state)
        else:
            st.error(f"Unknown step: {state['current_step']}")

    def render_requirements(self, state):
        st.markdown("### Requirements Input")
        requirements = st.text_area("Enter your requirements:", height=200)
        if st.button("Submit Requirements"):
            state["requirements"] = requirements
            state["current_step"] = "generate_user_stories"
            st.session_state.state["current_step"] = "generate_user_stories"
       
            
            

    def render_generate_user_stories(self, state):
        st.markdown("### Generated User Stories")
        if st.session_state.state['user_stories']!='':
            with st.expander("View User Stories"):
                st.markdown(st.session_state.state['user_stories'])
            if st.button("Continue to Product Owner Review"):
                state["current_step"] = "product_owner_review"
                st.session_state.state["current_step"] = "product_owner_review"
            # st.rerun()

    def render_product_owner_review(self, state):
        st.markdown("### Product Owner Review")
        with st.expander("View Current User Stories"):
            st.markdown(state.get("user_stories", ""))
        st.markdown("### Review Actions")
        feedback = st.text_area("Feedback:", height=150, key='Feedback')
        
        col1, col2 = st.columns(2)
        with col1:
            if st.button("🚀 Approve Design Phase", key="approve_design"):
                state["decision"] = "approved"
                state["current_step"] = "create_design_docs"
                st.session_state.user_decision = "approve"
                st.session_state.graph_stage = "resumed"
                st.rerun()
        with col2:
            if st.button("🔧 Request Revisions", key="request_revisions"):
                state["decision"] = "feedback"
                state["po_feedback"] = feedback
                state["current_step"] = "revise_user_stories"
                st.rerun()

    def render_create_design_docs(self, state):
        st.markdown("### Design Documents")
        with st.expander("View Design Documents"):
            st.markdown(state.get("design_docs", ""))
        if st.button("Proceed to Design Review"):
            state["current_step"] = "design_review"
            st.rerun()

    def render_design_review(self, state):
        st.markdown("### Design Review")
        with st.expander("View Design Documents"):
            st.markdown(state.get("design_docs", ""))
        st.markdown("### Review Actions")
        feedback = st.text_area("Design Feedback:", height=150, key="design_feedback")
        
        col1, col2 = st.columns(2)
        with col1:
            if st.button("🚀 Approve Implementation", key="approve_implementation"):
                state["decision"] = "approved"
                state["current_step"] = "generate_code"
                st.session_state.user_decision = "approve"
                st.session_state.graph_stage = "resumed"
                st.rerun()
        with col2:
            if st.button("🔧 Request Design Changes", key="request_design_changes"):
                state["decision"] = "feedback"
                state["design_feedback"] = feedback
                state["current_step"] = "create_design_docs"
                st.rerun()

    def render_generate_code(self, state):
        st.markdown("### Generated Code")
        with st.expander("View Code Implementation"):
            st.code(state.get("code", ""), language='python')
        if st.button("Proceed to Code Review"):
            state["current_step"] = "code_review"
            st.rerun()

    def render_code_review(self, state):
        st.markdown("### Code Review")
        with st.expander("View Current Code"):
            st.code(state.get("code", ""), language='python')
        st.markdown("### Review Actions")
        feedback = st.text_area("Code Feedback:", height=150, key="code_feedback")
        
        col1, col2 = st.columns(2)
        with col1:
            if st.button("🚀 Approve for Security Review", key="approve_security"):
                state["decision"] = "approved"
                state["current_step"] = "security_review"
                st.session_state.user_decision = "approve"
                st.session_state.graph_stage = "resumed"
                st.rerun()
        with col2:
            if st.button("🔧 Request Code Fixes", key="request_code_fixes"):
                state["decision"] = "feedback"
                state["review_feedback"] = feedback
                state["current_step"] = "fix_code_after_code_review"
                st.rerun()

    def render_security_review(self, state):
        st.markdown("### Security Review")
        with st.expander("View Code for Security Audit"):
            st.code(state.get("code", ""), language='python')
        st.markdown("### Security Findings")
        feedback = st.text_area("Security Feedback:", height=150, key="security_feedback")
        
        col1, col2 = st.columns(2)
        with col1:
            if st.button("🚀 Approve for Testing", key="approve_testing"):
                state["decision"] = "approved"
                state["current_step"] = "write_test_cases"
                st.session_state.user_decision = "approve"
                st.session_state.graph_stage = "resumed"
                st.rerun()
        with col2:
            if st.button("🔧 Request Security Fixes", key="request_security_fixes"):
                state["decision"] = "feedback"
                state["security_feedback"] = feedback
                state["current_step"] = "fix_code_after_security"
                st.rerun()

    def render_write_test_cases(self, state):
        st.markdown("### Test Cases")
        with st.expander("View Test Cases"):
            st.markdown(state.get("test_cases", ""))
        if st.button("Proceed to Test Review"):
            state["current_step"] = "test_cases_review"
            st.rerun()

    def render_test_cases_review(self, state):
        st.markdown("### Test Cases Review")
        with st.expander("View Current Test Cases"):
            st.markdown(state.get("test_cases", ""))
        st.markdown("### Review Actions")
        feedback = st.text_area("Test Feedback:", height=150, key="test_feedback")
        
        col1, col2 = st.columns(2)
        with col1:
            if st.button("🚀 Final Approval", key="final_approve"):
                state["decision"] = "approved"
                state["current_step"] = "end"
                st.rerun()
        with col2:
            if st.button("🔧 Improve Test Cases", key="request_test_fixes"):
                state["decision"] = "feedback"
                state["test_feedback"] = feedback
                state["current_step"] = "fix_test_cases"
                st.rerun()

    def render_end(self, state):
        st.success("✅ SDLC Process Completed Successfully!")
        with st.expander("Requirements"):
            st.write(state.get("requirements", ""))
        with st.expander("User Stories"):
            st.write(state.get("user_stories", ""))
        with st.expander("Design Documents"):
            st.write(state.get("design_docs", ""))
        with st.expander("Design Feedback"):
            st.write(state.get("design_feedback", ""))
        with st.expander("Generated Code"):
            st.write(state.get("generate_code", ""))
        with st.expander("Test Cases"):
            st.write(state.get("test_cases", ""))
        with st.expander("PO Feedback"):
            st.write(state.get("po_feedback", ""))
        with st.expander("Review Feedback"):
            st.write(state.get("review_feedback", ""))
        with st.expander("Security Feedback"):
            st.write(state.get("security_feedback", ""))
        with st.expander("Test Feedback"):
            st.write(state.get("test_feedback", ""))
                
            
            
        with st.expander("Final Artifacts", expanded=True):
            st.markdown("### Download Assets")
            st.download_button("📥 Requirements", 
                            data=state.get("requirements", ""), 
                            file_name="requirements.md")
            st.download_button("📥 User Stories", 
                            data=state.get("user_stories", ""), 
                            file_name="user_stories.md")
            st.download_button("📥 Design Documents", 
                            data=state.get("design_docs", ""), 
                            file_name="design_docs.md")
            st.download_button("📥 Generated Code", 
                            data=state.get("code", ""), 
                            file_name="generated_code.md")
            st.download_button("📥 Test Cases", 
                            data=state.get("test_cases", ""), 
                            file_name="test_cases.md")
            st.download_button("📥 Product Owner Feedback", 
                            data=state.get("po_feedback", ""), 
                            file_name="po_feedback.md")
            st.download_button("📥 Design Feedback", 
                            data=state.get("design_feedback", ""), 
                            file_name="design_feedback.md")
            st.download_button("📥 Code Review Feedback", 
                            data=state.get("review_feedback", ""), 
                            file_name="code_review_feedback.md")
            st.download_button("📥 Security Feedback", 
                            data=state.get("security_feedback", ""), 
                            file_name="security_feedback.md")
            st.download_button("📥 Test Cases Feedback", 
                            data=state.get("test_feedback", ""), 
                            file_name="test_feedback.md")

                
        if st.button("🔄 Restart Process"):
            st.session_state.clear()
            st.rerun()