Update app.py

app.py

@@ -22,6 +22,15 @@ EXAMPLE_QUESTIONS = [
     "How can I make this code more readable?"
 ]
 
+LANGUAGE_KEYWORDS = {
+    "Python": ["def ", "import ", "self", "print(", "lambda", "None"],
+    "JavaScript": ["function ", "console.log", "var ", "let ", "const ", "=>"],
+    "TypeScript": ["interface ", "type ", ": string", ": number", "export ", "import "],
+    "Java": ["public class", "System.out.println", "void main", "import java.", "new "],
+    "C++": ["#include", "std::", "cout <<", "cin >>", "int main(", "using namespace"],
+    "C#": ["using System;", "namespace ", "public class", "Console.WriteLine", "static void Main"]
+}
+
 # --- API STUBS ---
 def call_groq_api(prompt, model="llama3-70b-8192"):
     # Replace with actual Groq API call
@@ -51,10 +60,8 @@ def call_blackbox_agent(messages):
 
 # --- UTILS ---
 def code_matches_language(code, language):
-    # Simple heuristic, can be improved
-    if language.lower() in code.lower():
-        return True
-    return True  # For demo, always True
+    keywords = LANGUAGE_KEYWORDS.get(language, [])
+    return any(kw in code for kw in keywords)
 
 def calculate_code_complexity(code):
     # Dummy complexity metric
@@ -86,6 +93,16 @@ def is_coding_question(question):
     except Exception:
         return False
 
+def get_explanation_prompt(code, programming_language, skill_level, user_role, explanation_language, question=None):
+    lang_instruction = ""
+    if explanation_language != "English":
+        lang_instruction = f" Respond in {explanation_language}."
+    if question:
+        return f"{question}\n\nCode:\n{code}\n{lang_instruction}"
+    return (
+        f"Explain this {programming_language} code for a {skill_level} {user_role}.{lang_instruction}\n{code}"
+    )
+
 # --- STREAMLIT APP ---
 st.set_page_config(page_title="Code Workflows", layout="wide")
 st.title("Code Genie")
@@ -127,9 +144,8 @@ elif page == "Code Workflows":
             st.error(f"Language mismatch. Please check your code and language selection.")
         else:
             with st.spinner("Running AI Workflow..."):
-                # Simulate workflow steps
                 steps = [
-                    ("Explain", call_groq_api(f"Explain this {programming_language} code for a {skill_level} {user_role} in {explanation_language}:\n{code_input}")),
+                    ("Explain", call_groq_api(get_explanation_prompt(code_input, programming_language, skill_level, user_role, explanation_language))),
                     ("Refactor", call_blackbox_agent([
                         {"role": "system", "content": "You are a helpful coding assistant."},
                         {"role": "user", "content": f"Refactor this {programming_language} code: {code_input}"}
@@ -186,7 +202,8 @@ elif page == "Semantic Search":
             st.error(f"Language mismatch. Please check your code and language selection.")
         else:
             with st.spinner("Running Semantic Search..."):
-                answer = call_groq_api(f"{question}\n\nCode:\n{code_input}")
+                prompt = get_explanation_prompt(code_input, programming_language, skill_level, user_role, explanation_language, question=question)
+                answer = call_groq_api(prompt)
                 st.success("Answer:")
                 st.write(answer)
 
@@ -201,6 +218,8 @@ elif page == "Code Comment Generator":
     if st.button("Generate Comments"):
         if not code_input.strip():
             st.error("Please paste or upload your code.")
+        elif not code_matches_language(code_input, programming_language):
+            st.error(f"Language mismatch. Please check your code and language selection.")
         else:
             with st.spinner("Generating commented code..."):
                 prompt = (