Update app.py

app.py

@@ -11,34 +11,35 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
 # Streamlit UI
-st.title("AI Code Generator & Debugger")
+st.title("🤖 AI Code Generator & Debugger 🖥️")
 st.write("Generate and debug code in multiple programming languages.")
 
-language = st.selectbox("Select Programming Language", ["Python", "Java", "JavaScript", "PHP", "C++", "C#"])
-mode = st.radio("Select Mode", ["Generate Code", "Debug Code"])
-user_input = st.text_area("Enter your prompt or code snippet")
-submit = st.button("Submit")
+language = st.selectbox("🛠️ Select Programming Language", ["Python", "Java", "JavaScript", "PHP", "C++", "C#"])
+mode = st.radio("🎯 Select Mode", ["Generate Code", "Debug Code"])
+user_input = st.text_area("💡 Enter your prompt or code snippet")
+submit = st.button("🚀 Submit")
 
 def get_groq_response(prompt):
+    """Fetch response from Groq AI model."""
     completion = client.chat.completions.create(
-        model="deepseek-r1-distill-llama-70b",
-        messages=[{"role": "system", "content": "You are an expert AI code assistant."},
-                  {"role": "user", "content": prompt}],
+        model="llama-3.3-70b-versatile",
+        messages=[
+            {"role": "system", "content": "You are an expert AI code assistant."},
+            {"role": "user", "content": prompt}
+        ],
         temperature=0.6,
         max_completion_tokens=4096,
         top_p=0.95,
-        stream=True,
+        stream=False,
         stop=None,
     )
-    return "".join(chunk.choices[0].delta.content or "" for chunk in completion)
+    return completion.choices[0].message.content.strip()
 
 if submit:
     if user_input.strip():
-        if any(lang.lower() in user_input.lower() for lang in ["python", "java", "javascript", "php", "c++", "c#"]):
-            prompt = f"{mode} {language} code: {user_input}"
-            response = get_groq_response(prompt)
-            st.code(response, language=language.lower())
-        else:
-            st.error("Invalid question. Please try again with a relevant coding prompt.")
+        prompt = f"{mode} {language} code: {user_input}"
+        response = get_groq_response(prompt)
+        st.subheader("📌 AI Generated Response:")
+        st.code(response, language=language.lower())
     else:
-        st.warning("Please enter a prompt or code snippet.")
+        st.warning("⚠️ Please enter a prompt or code snippet.")