Update app.py

app.py

@@ -3,35 +3,55 @@ from groq import Groq
 import requests
 
 # =======================
-# API Clients
+# API Clients Initialization
 # =======================
-client = Groq(api_key=st.secrets["GROQ_API_KEY"])
-HF_API_KEY = st.secrets["HF_API_KEY"]
+groq_client = None
+groq_status = "❌"
+if "GROQ_API_KEY" in st.secrets:
+    try:
+        groq_client = Groq(api_key=st.secrets["GROQ_API_KEY"])
+        # simple test request
+        groq_client.models.list()
+        groq_status = "✅"
+    except Exception:
+        groq_client = None
+        groq_status = "❌"
+
+huggingface_api_key = st.secrets.get("HF_API_KEY", None)
+hf_status = "❌"
+if huggingface_api_key:
+    try:
+        headers = {"Authorization": f"Bearer {huggingface_api_key}"}
+        response = requests.get("https://api-inference.huggingface.co/status", headers=headers)
+        if response.status_code in [200, 401, 403]:  # API exists
+            hf_status = "✅"
+    except Exception:
+        hf_status = "❌"
+
 
 # =======================
 # Sidebar Settings
 # =======================
 st.sidebar.title("⚙️ Settings")
-provider_choice = st.sidebar.radio(
-    "Choose Provider",
-    ["Groq", "Hugging Face"]
+st.sidebar.markdown(f"**Groq API Status:** {groq_status}")
+st.sidebar.markdown(f"**HuggingFace API Status:** {hf_status}")
+
+api_priority = st.sidebar.radio(
+    "Choose API Priority",
+    ["Groq First", "HuggingFace First"],
+    index=0
 )
 
-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"]
-    )
-else:
-    model_choice = st.sidebar.selectbox(
-        "Choose HF Model",
-        ["mistralai/Mixtral-8x7B-Instruct-v0.1", "tiiuae/falcon-7b-instruct"]
-    )
+model_choice = st.sidebar.selectbox(
+    "Choose Model",
+    ["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "mixtral-8x7b-32768"]
+)
 
 st.title("🤖 CodeCraft AI - Mini Copilot (Chat Edition)")
 
+
 # =======================
-# Session state for chats
+# Session State
 # =======================
 if "generate_chat" not in st.session_state:
     st.session_state.generate_chat = []
@@ -42,55 +62,80 @@ if "explain_chat" not in st.session_state:
 
 
 # =======================
-# Helper functions
+# Helper Functions
 # =======================
-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):
-    headers = {"Authorization": f"Bearer {HF_API_KEY}"}
-    payload = {"inputs": prompt}
-    response = requests.post(
-        f"https://api-inference.huggingface.co/models/{model_choice}",
-        headers=headers,
-        json=payload,
-        timeout=60
-    )
-    if response.status_code == 200:
-        result = response.json()
-        if isinstance(result, list) and "generated_text" in result[0]:
-            return result[0]["generated_text"]
+def call_groq(system_prompt, chat_history):
+    """Call Groq API with error handling"""
+    if not groq_client:
+        return None
+    try:
+        response = groq_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:
+        st.warning(f"⚠️ Groq API error: {e}")
+        return None
+
+
+def call_huggingface(system_prompt, chat_history):
+    """Call Hugging Face Inference API with error handling"""
+    if not huggingface_api_key:
+        return None
+    try:
+        headers = {"Authorization": f"Bearer {huggingface_api_key}"}
+        payload = {
+            "inputs": "\n".join([msg for _, msg in chat_history]),
+            "parameters": {"temperature": 0.5, "max_new_tokens": 500}
+        }
+        response = requests.post(
+            f"https://api-inference.huggingface.co/models/{model_choice}",
+            headers=headers,
+            json=payload
+        )
+        if response.status_code == 200:
+            data = response.json()
+            if isinstance(data, list) and "generated_text" in data[0]:
+                return data[0]["generated_text"]
+            return str(data)
         else:
-            return str(result)
-    return f"⚠️ HF Error: {response.text}"
+            st.warning(f"⚠️ Hugging Face error: {response.text}")
+            return None
+    except Exception as e:
+        st.warning(f"⚠️ Hugging Face exception: {e}")
+        return None
 
 
-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)
+def get_ai_response(system_prompt, chat_history):
+    """Get AI response using priority + fallback"""
+    if api_priority == "Groq First":
+        ai_msg = call_groq(system_prompt, chat_history)
+        if ai_msg is None:
+            ai_msg = call_huggingface(system_prompt, chat_history)
+    else:  # HuggingFace First
+        ai_msg = call_huggingface(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
 
 
 # =======================
-# Chat UI
+# Chat UI Function
 # =======================
 def chat_ui(tab_name, chat_history, system_prompt, input_key):
+    """Reusable chat UI for each tab with fixed bottom input bar"""
+
     st.subheader(tab_name)
 
     # --- Chat history display ---
-    with st.container():
+    chat_container = st.container()
+    with chat_container:
         for role, msg in chat_history:
             if role == "user":
                 with st.chat_message("user"):
@@ -109,26 +154,20 @@ def chat_ui(tab_name, chat_history, system_prompt, input_key):
                     else:
                         st.write(msg)
 
-    # --- 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")
+    # --- Chat input ---
+    user_input = st.chat_input("Type your message...", key=input_key)
 
-    # --- Handle input ---
-    if send_btn and user_input.strip():
-        chat_history.append(("user", user_input.strip()))
+    if user_input:
+        # Save user input
+        st.session_state[input_key + "_last"] = user_input
+        chat_history.append(("user", user_input))
 
-        with st.spinner("Thinking..."):
-            ai_msg = get_ai_response(chat_history, system_prompt)
+        # --- Get AI reply with fallback ---
+        with st.spinner("🤔 Thinking..."):
+            ai_msg = get_ai_response(system_prompt, chat_history)
 
         chat_history.append(("assistant", ai_msg))
-        st.session_state[input_key] = ""  # clear input
+
         st.rerun()
 
 
@@ -161,8 +200,9 @@ with tab3:
         input_key="explain_input"
     )
 
+
 # =======================
 # Footer
 # =======================
 st.markdown("---")
-st.caption("🚀 Built for Hackathons using Streamlit + Hugging Face + Groq")
+st.caption("✨ CodeCraft may make mistakes, so double-check important info.")