Update app.py

app.py

@@ -13,16 +13,19 @@ HF_API_KEY = st.secrets["HF_API_KEY"]
 # =======================
 st.sidebar.title("⚙️ Settings")
 provider_choice = st.sidebar.radio(
-    "Choose Provider", ["Groq", "Hugging Face"]
+    "Choose Provider",
+    ["Groq", "Hugging Face"]
 )
 
 if provider_choice == "Groq":
     model_choice = st.sidebar.selectbox(
-        "Choose Groq Model", ["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "mixtral-8x7b-32768"]
+        "Choose Groq Model",
+        ["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "mixtral-8x7b-32768"]
     )
 else:
     model_choice = st.sidebar.selectbox(
-        "Choose HF Model", ["mistralai/Mixtral-8x7B-Instruct-v0.1", "tiiuae/falcon-7b-instruct"]
+        "Choose HF Model",
+        ["mistralai/Mixtral-8x7B-Instruct-v0.1", "tiiuae/falcon-7b-instruct"]
     )
 
 st.title("🤖 CodeCraft AI - Mini Copilot (Chat Edition)")
@@ -37,21 +40,21 @@ if "debug_chat" not in st.session_state:
 if "explain_chat" not in st.session_state:
     st.session_state.explain_chat = []
 
+
 # =======================
 # Helper functions
 # =======================
-def call_groq(chat_history: list[tuple[str, str]], system_prompt: str) -> str:
-    try:
-        response = client.chat.completions.create(
-            model=model_choice,
-            messages=[{"role": "system", "content": system_prompt}] + [{"role": role, "content": msg} for role, msg in chat_history],
-            temperature=0.4
-        )
-        return response.choices[0].message.content
-    except Exception as e:
-        return f"⚠️ Groq Error: {str(e)}"
+def call_groq(chat_history, system_prompt):
+    response = client.chat.completions.create(
+        model=model_choice,
+        messages=[{"role": "system", "content": system_prompt}]
+                 + [{"role": role, "content": msg} for role, msg in chat_history],
+        temperature=0.4
+    )
+    return response.choices[0].message.content
 
-def call_hf(prompt: str) -> str:
+
+def call_hf(prompt):
     headers = {"Authorization": f"Bearer {HF_API_KEY}"}
     payload = {"inputs": prompt}
     response = requests.post(
@@ -68,22 +71,24 @@ def call_hf(prompt: str) -> str:
             return str(result)
     return f"⚠️ HF Error: {response.text}"
 
-def get_ai_response(chat_history: list[tuple[str, str]], system_prompt: str) -> str:
+
+def get_ai_response(chat_history, system_prompt):
     if provider_choice == "Groq":
         return call_groq(chat_history, system_prompt)
     else:
+        # Convert history into plain text prompt for Hugging Face
         prompt = system_prompt + "\n\n"
         for role, msg in chat_history:
             prompt += f"{role.upper()}: {msg}\n"
         return call_hf(prompt)
 
 
-
 # =======================
 # Chat UI
 # =======================
 def chat_ui(tab_name, chat_history, system_prompt, input_key):
     st.subheader(tab_name)
+
     # --- Chat history display ---
     with st.container():
         for role, msg in chat_history:
@@ -106,20 +111,27 @@ def chat_ui(tab_name, chat_history, system_prompt, input_key):
 
     # --- Input bar + send button in one row ---
     col1, col2 = st.columns([10, 1])
-with col1:
-    user_input = st.text_input("Type your message...", key=input_key, label_visibility="collapsed")
-with col2:
-    send_btn = st.button("➤", key=input_key + "_send")
+    with col1:
+        user_input = st.text_input(
+            "Type your message...",
+            key=input_key,
+            label_visibility="collapsed"
+        )
+    with col2:
+        send_btn = st.button("➤", key=input_key + "_send")
+
     # --- Handle input ---
     if send_btn and user_input.strip():
         chat_history.append(("user", user_input.strip()))
+
         with st.spinner("Thinking..."):
             ai_msg = get_ai_response(chat_history, system_prompt)
-            chat_history.append(("assistant", ai_msg))
+
+        chat_history.append(("assistant", ai_msg))
+        st.session_state[input_key] = ""  # clear input
         st.rerun()
 
 
-        
 # =======================
 # Tabs
 # =======================