Update app.py

app.py

@@ -29,48 +29,76 @@ if huggingface_api_key:
 
 
 # =======================
-# Sidebar (Scrollable + Features)
+# Session State
 # =======================
-st.sidebar.title("⚙️ Settings")
-
-with st.sidebar:
-    st.markdown(f"**Groq API Status:** {groq_status}")
-    st.markdown(f"**HuggingFace API Status:** {hf_status}")
-
-    with st.expander("🔧 API Settings", expanded=True):
-        api_priority = st.radio(
-            "Choose API Priority",
-            ["Groq First", "HuggingFace First"],
-            index=0
-        )
-        model_choice = st.selectbox(
-            "Choose Model",
-            ["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "mixtral-8x7b-32768"]
-        )
+if "chats" not in st.session_state:
+    st.session_state.chats = {
+        "💡 Generate Code": [],
+        "🛠 Debug Code": [],
+        "📘 Explain Code": []
+    }
+if "active_chat" not in st.session_state:
+    st.session_state.active_chat = "💡 Generate Code"
 
-    st.markdown("---")
-    st.markdown("### 📂 Chats")
 
-    # Sidebar dots + chat list
-    if "custom_chats" not in st.session_state:
-        st.session_state.custom_chats = {"Chat 1": []}
-
-    for i, (chat_name, chat_history) in enumerate(st.session_state.custom_chats.items(), start=1):
-        st.markdown(f"• {chat_name}")
+# =======================
+# Sidebar - Chat Manager
+# =======================
+st.sidebar.title("⚙️ Settings")
+st.sidebar.markdown(f"**Groq API Status:** {groq_status}")
+st.sidebar.markdown(f"**HuggingFace API Status:** {hf_status}")
 
-    if st.button("+ New Chat"):
-        new_chat_name = f"Chat {len(st.session_state.custom_chats) + 1}"
-        st.session_state.custom_chats[new_chat_name] = []
+api_priority = st.sidebar.radio(
+    "Choose API Priority",
+    ["Groq First", "HuggingFace First"],
+    index=0
+)
 
+model_choice = st.sidebar.selectbox(
+    "Choose Model",
+    ["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "mixtral-8x7b-32768"]
+)
 
-# =======================
-# Main Title
-# =======================
-st.title("🤖 CodeCraft AI - Mini Copilot (Chat Edition)")
+st.sidebar.markdown("---")
+st.sidebar.subheader("💬 My Chats")
+
+# Scrollable sidebar with chats
+with st.sidebar.container():
+    for chat_name in list(st.session_state.chats.keys()):
+        col1, col2 = st.sidebar.columns([4, 1])
+        if col1.button(chat_name, key=f"open_{chat_name}"):
+            st.session_state.active_chat = chat_name
+
+        with col2:
+            dots = st.button("⋮", key=f"menu_{chat_name}")
+            if dots:
+                action = st.radio(
+                    f"Action for {chat_name}",
+                    ["Cancel", "Rename", "Delete"],
+                    key=f"action_{chat_name}",
+                    horizontal=True
+                )
+                if action == "Rename":
+                    new_name = st.text_input("New name", key=f"rename_{chat_name}")
+                    if new_name and new_name not in st.session_state.chats:
+                        st.session_state.chats[new_name] = st.session_state.chats.pop(chat_name)
+                        st.session_state.active_chat = new_name
+                        st.rerun()
+                elif action == "Delete":
+                    st.session_state.chats.pop(chat_name)
+                    if st.session_state.active_chat == chat_name:
+                        st.session_state.active_chat = list(st.session_state.chats.keys())[0]
+                    st.rerun()
+
+if st.sidebar.button("➕ Add Chat"):
+    new_chat_name = f"Chat {len(st.session_state.chats) + 1}"
+    st.session_state.chats[new_chat_name] = []
+    st.session_state.active_chat = new_chat_name
+    st.rerun()
 
 
 # =======================
-# Helper Functions
+# API Calls
 # =======================
 def call_groq(system_prompt, chat_history):
     if not groq_client:
@@ -125,38 +153,33 @@ def get_ai_response(system_prompt, chat_history):
         if ai_msg is None:
             ai_msg = call_groq(system_prompt, chat_history)
 
-    if ai_msg is None:
-        return "❌ Both APIs failed. Please check your API keys or try again later."
-    return ai_msg
+    return ai_msg or "❌ Both APIs failed. Please check your API keys."
 
 
 # =======================
-# Chat UI Function
+# Chat UI
 # =======================
-def chat_ui(tab_name, chat_history, system_prompt, input_key):
-    st.subheader(tab_name)
+def chat_ui(chat_name, system_prompt, input_key):
+    st.subheader(chat_name)
+
+    chat_history = st.session_state.chats[chat_name]
 
     chat_container = st.container()
     with chat_container:
         for role, msg in chat_history:
-            if role == "user":
-                with st.chat_message("user"):
+            with st.chat_message(role):
+                if role == "assistant" and "```" in msg:
+                    parts = msg.split("```")
+                    for i, part in enumerate(parts):
+                        if i % 2 == 1:
+                            lang, *code_lines = part.split("\n")
+                            code = "\n".join(code_lines)
+                            st.code(code, language=lang if lang else "python")
+                        else:
+                            st.write(part)
+                else:
                     st.write(msg)
-            else:
-                with st.chat_message("assistant"):
-                    if "```" in msg:
-                        parts = msg.split("```")
-                        for i, part in enumerate(parts):
-                            if i % 2 == 1:
-                                lang, *code_lines = part.split("\n")
-                                code = "\n".join(code_lines)
-                                st.code(code, language=lang if lang else "python")
-                            else:
-                                st.write(part)
-                    else:
-                        st.write(msg)
-
-    # --- Fixed input at bottom ---
+
     user_input = st.chat_input("Type your message...", key=input_key)
 
     if user_input:
@@ -168,67 +191,23 @@ def chat_ui(tab_name, chat_history, system_prompt, input_key):
 
 
 # =======================
-# Tabs
+# Main Chat Area
 # =======================
-tab1, tab2, tab3 = st.tabs(["💡 Generate Code", "🛠 Debug Code", "📘 Explain Code"])
-
-with tab1:
-    if "generate_chat" not in st.session_state:
-        st.session_state.generate_chat = []
-    chat_ui(
-        "💡 Generate Code",
-        st.session_state.generate_chat,
-        "You are a helpful coding assistant. Generate correct code first, then a short simple explanation.",
-        input_key="generate_input"
-    )
-
-with tab2:
-    if "debug_chat" not in st.session_state:
-        st.session_state.debug_chat = []
-    chat_ui(
-        "🛠 Debug Code",
-        st.session_state.debug_chat,
-        "You are an expert code debugger. Fix errors and give corrected code, then explain what changed and why in simple terms.",
-        input_key="debug_input"
-    )
-
-with tab3:
-    if "explain_chat" not in st.session_state:
-        st.session_state.explain_chat = []
-    chat_ui(
-        "📘 Explain Code",
-        st.session_state.explain_chat,
-        "You are a teacher that explains code in simple words. The user pastes code, and you explain step by step.",
-        input_key="explain_input"
-    )
+system_prompts = {
+    "💡 Generate Code": "You are a helpful coding assistant...",
+    "🛠 Debug Code": "You are an expert code debugger...",
+    "📘 Explain Code": "You are a teacher explaining code..."
+}
+
+active = st.session_state.active_chat
+if active in system_prompts:
+    chat_ui(active, system_prompts[active], input_key=f"{active}_input")
+else:
+    chat_ui(active, "You are a helpful assistant.", input_key=f"{active}_input")
 
 
 # =======================
-# Footer (Fixed bottom)
+# Footer
 # =======================
-st.markdown(
-    """
-    <style>
-        .block-container {
-            padding-bottom: 80px; /* leave space for footer */
-        }
-        footer {
-            visibility: hidden;
-        }
-        .custom-footer {
-            position: fixed;
-            bottom: 0;
-            left: 0;
-            width: 100%;
-            text-align: center;
-            background-color: white;
-            padding: 8px;
-            border-top: 1px solid #ddd;
-        }
-    </style>
-    <div class="custom-footer">
-        ✨ CodeCraft may make mistakes. Always check important info.
-    </div>
-    """,
-    unsafe_allow_html=True
-)
+st.markdown("---")
+st.caption("✨ CodeCraft may make mistakes. Check important info.")