Update app.py

app.py

@@ -152,15 +152,16 @@ elif page == "Code Workflows":
         else:
             with st.spinner("Running AI Workflow..."):
                 lang_instruction = f" Respond in {explanation_language}." if explanation_language != "English" else ""
+                role_level_instruction = f" The user is a {skill_level} {user_role}."
                 steps = [
                     ("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}{lang_instruction}"}
+                        {"role": "user", "content": f"Refactor this {programming_language} code for a {skill_level} {user_role}: {code_input}{lang_instruction}"}
                     ])),
-                    ("Review", call_groq_api(f"Review this {programming_language} code for errors and improvements: {code_input}")),
-                    ("ErrorDetection", call_groq_api(f"Find bugs in this {programming_language} code: {code_input}")),
-                    ("TestGeneration", call_groq_api(f"Generate tests for this {programming_language} code: {code_input}")),
+                    ("Review", call_groq_api(f"Review this {programming_language} code for errors and improvements for a {skill_level} {user_role}: {code_input}{lang_instruction}")),
+                    ("ErrorDetection", call_groq_api(f"Find bugs in this {programming_language} code for a {skill_level} {user_role}: {code_input}{lang_instruction}")),
+                    ("TestGeneration", call_groq_api(f"Generate tests for this {programming_language} code for a {skill_level} {user_role}: {code_input}{lang_instruction}")),
                 ]
                 timeline = []
                 for step, output in steps:
@@ -279,10 +280,10 @@ elif page == "Code Comment Generator":
             st.error(f"Language mismatch. Please check your code and language selection.")
         else:
             with st.spinner("Generating commented code..."):
+                lang_instruction = f" Respond in {explanation_language}." if explanation_language != "English" else ""
+                role_level_instruction = f" The user is a {skill_level} {user_role}."
                 prompt = (
-                    f"Add clear, helpful comments to this {programming_language} code. "
-                    "Keep the code unchanged except for adding comments. "
-                    "Return the full code with comments:\n\n"
+                    f"Add clear, helpful comments to this {programming_language} code for a {skill_level} {user_role}.{lang_instruction}\n\n"
                     f"{code_input}"
                 )
                 commented_code = call_blackbox_agent([
@@ -297,7 +298,10 @@ elif page == "Code Comment Generator":
                     "timestamp": str(datetime.datetime.now()),
                     "user_code": code_input,
                     "params": {
-                        "language": programming_language
+                        "language": programming_language,
+                        "skill": skill_level,
+                        "role": user_role,
+                        "explanation_language": explanation_language
                     },
                     "commented_code": commented_code
                 })