Update app.py

app.py

@@ -1,138 +1,113 @@
 import streamlit as st
-from huggingface_hub import InferenceClient
 from groq import Groq
 
-# =======================
-# API Keys
-# =======================
-HF_API_TOKEN = st.secrets.get("HF_API_TOKEN", None)
-GROQ_API_KEY = st.secrets.get("GROQ_API_KEY", None)
-
-# =======================
-# Model Lists
-# =======================
-HF_MODELS = [
-    "bigcode/starcoder2-3b",
-    "mistralai/Mistral-7B-Instruct-v0.2",
-    "tiiuae/falcon-7b-instruct"
-]
-
-GROQ_MODELS = [
-    "llama-3.1-8b-instant",
-    "llama-3.3-70b-versatile",
-    "openai/gpt-oss-20b",
-    "meta-llama/llama-guard-4-12b"
-]
-
-# =======================
-# Query Functions
-# =======================
-def query_hf(model, prompt):
-    try:
-        client = InferenceClient(token=HF_API_TOKEN)
-        response = client.text_generation(
-            model=model,
-            prompt=prompt,
-            max_new_tokens=600,
-            temperature=0.7
-        )
-        return response
-    except Exception as e:
-        return f"⚠️ Hugging Face Error: {e}"
-
-def query_groq(model, prompt):
-    try:
-        client = Groq(api_key=GROQ_API_KEY)
-        resp = client.chat.completions.create(
-            model=model,
-            messages=[{"role": "user", "content": prompt}],
-            temperature=0.7,
-            max_tokens=600
-        )
-        return resp.choices[0].message.content
-    except Exception as e:
-        return f"⚠️ Groq Error: {e}"
-
-# =======================
-# Streamlit UI
-# =======================
-st.set_page_config(page_title="CodeCraft AI", page_icon="⚡", layout="wide")
-st.title("⚡ CodeCraft AI – Mini Copilot for Hackathons")
+# Initialize Groq client
+client = Groq(api_key=st.secrets["GROQ_API_KEY"])
+
+# Sidebar for model selection
+st.sidebar.title("⚙️ Settings")
+model_choice = st.sidebar.selectbox(
+    "Choose AI Model",
+    ["llama-3.1-8b-instant", "llama-3.1-70b-versatile", "mixtral-8x7b-32768"]
+)
+
+st.title("🤖 CodeCraft AI - Mini Copilot")
+
+# Tabs
+tab1, tab2, tab3 = st.tabs(["💡 Generate Code", "🛠 Debug Code", "📘 Explain Code"])
+
+# ------------------------- Generate Code -------------------------
+with tab1:
+    st.subheader("💡 Generate Code from Statement")
+    statement = st.text_area("Enter your problem statement (e.g., 'Build a calculator in Python')")
+
+    if st.button("Generate Code"):
+        if statement.strip():
+            with st.spinner("Generating code..."):
+                response = client.chat.completions.create(
+                    model=model_choice,
+                    messages=[
+                        {"role": "system", "content": "You are a helpful coding assistant."},
+                        {"role": "user", "content": f"Generate correct code for: {statement}. "
+                                                    f"First show the code block clearly. "
+                                                    f"Then give a short simple explanation."}
+                    ],
+                    temperature=0.3
+                )
+                result = response.choices[0].message.content
 
-st.markdown("Choose a task below: Generate Code, Debug, or Explain.")
-
-backend = st.radio("Choose Backend", ["Hugging Face", "Groq"])
-
-if backend == "Hugging Face":
-    model_choice = st.selectbox("Select Hugging Face Model", HF_MODELS)
-else:
-    model_choice = st.selectbox("Select Groq Model", GROQ_MODELS)
-
-tabs = st.tabs(["💻 Generate Code", "🐞 Debug Code", "📖 Explain Code"])
+            # Split into code + explanation
+            if "```" in result:
+                parts = result.split("```")
+                code = parts[1] if len(parts) > 1 else ""
+                explanation = "".join(parts[2:]) if len(parts) > 2 else result
+            else:
+                code, explanation = result, ""
 
-# -----------------------
-# Generate Code Tab
-# -----------------------
-with tabs[0]:
-    prompt = st.text_area("💡 Enter your problem statement", height=200, key="gen_input")
-    if st.button("🚀 Generate Code"):
-        if not prompt.strip():
-            st.warning("Please enter something.")
+            # Show code (with copy button provided by Streamlit automatically)
+            st.code(code, language="python")
+            # Show explanation (plain text, no copy)
+            st.markdown("### 📘 Explanation")
+            st.write(explanation.strip())
         else:
-            with st.spinner("AI is generating code..."):
-                output = query_hf(model_choice, prompt) if backend == "Hugging Face" else query_groq(model_choice, prompt)
-            st.subheader("✅ Generated Code")
-            st.code(output, language="python")
-
-# -----------------------
-# Debug Tab
-# -----------------------
-with tabs[1]:
-    buggy_code = st.text_area("🐞 Paste your buggy code here", height=200, key="debug_input")
-    if st.button("🔧 Debug Code"):
-        if not buggy_code.strip():
-            st.warning("Please paste some code.")
-        else:
-            debug_prompt = f"""
-You are a Python code debugger. The user has provided buggy code.
-1. First, output the corrected version of the code ONLY inside a code block.
-2. Then, after the code block, explain clearly what needed to change and what was changed to fix the code.
-Do not list fixes as Fix 1, Fix 2. Just a flowing explanation.
-
-Buggy code:
-{buggy_code}
-"""
-            with st.spinner("AI is debugging..."):
-                output = query_hf(model_choice, debug_prompt) if backend == "Hugging Face" else query_groq(model_choice, debug_prompt)
-
-            # Split into code + explanation
-            if "```" in output:
-                try:
-                    corrected = output.split("```")[1].replace("python", "").strip()
-                    explanation = output.split("```")[-1].strip()
-                    st.subheader("✅ Corrected Code")
-                    st.code(corrected, language="python")
-                    st.subheader("📘 Explanation")
-                    st.write(explanation)
-                except:
-                    st.write(output)
+            st.warning("⚠️ Please enter a statement.")
+
+# ------------------------- Debug Code -------------------------
+with tab2:
+    st.subheader("🛠 Debug Your Code")
+    user_code = st.text_area("Paste your buggy code here")
+
+    if st.button("Debug Code"):
+        if user_code.strip():
+            with st.spinner("Analyzing and fixing your code..."):
+                response = client.chat.completions.create(
+                    model=model_choice,
+                    messages=[
+                        {"role": "system", "content": "You are an expert code debugger."},
+                        {"role": "user", "content": f"Here is code: {user_code}. "
+                                                    f"Fix errors and give corrected code. "
+                                                    f"Then explain simply what I needed to change and why. "
+                                                    f"Do not use 'fix 1, fix 2' format, just a simple explanation."}
+                    ],
+                    temperature=0.2
+                )
+                result = response.choices[0].message.content
+
+            if "```" in result:
+                parts = result.split("```")
+                code = parts[1] if len(parts) > 1 else ""
+                explanation = "".join(parts[2:]) if len(parts) > 2 else result
             else:
-                st.write(output)
-
-# -----------------------
-# Explain Tab
-# -----------------------
-with tabs[2]:
-    code_to_explain = st.text_area("📖 Paste code to explain", height=200, key="explain_input")
-    if st.button("📝 Explain Code"):
-        if not code_to_explain.strip():
-            st.warning("Please paste some code.")
+                code, explanation = result, ""
+
+            st.code(code, language="python")
+            st.markdown("### 📘 Explanation of Fixes")
+            st.write(explanation.strip())
+        else:
+            st.warning("⚠️ Please paste your code.")
+
+# ------------------------- Explain Code -------------------------
+with tab3:
+    st.subheader("📘 Explain Your Code")
+    code_to_explain = st.text_area("Paste your code to understand it")
+
+    if st.button("Explain Code"):
+        if code_to_explain.strip():
+            with st.spinner("Explaining your code..."):
+                response = client.chat.completions.create(
+                    model=model_choice,
+                    messages=[
+                        {"role": "system", "content": "You are a teacher that explains code in simple terms."},
+                        {"role": "user", "content": f"Explain this code in very simple words:\n{code_to_explain}"}
+                    ],
+                    temperature=0.5
+                )
+                explanation = response.choices[0].message.content
+
+            st.markdown("### 📘 Explanation")
+            st.write(explanation.strip())
         else:
-            explain_prompt = f"Explain this Python code in simple steps:\n{code_to_explain}"
-            with st.spinner("AI is explaining..."):
-                output = query_hf(model_choice, explain_prompt) if backend == "Hugging Face" else query_groq(model_choice, explain_prompt)
-            st.subheader("📘 Explanation")
-            st.write(output)
+            st.warning("⚠️ Please paste your code.")
 
 
 # =======================