Update app.py

app.py

@@ -1,67 +1,93 @@
+# app.py
 import os
 import streamlit as st
 from groq import Groq
 
-api = "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"  # Replace with your actual API key
+# Configuration
+PRIMARY_MODEL = "qwen-2.5-coder-32b"
+BACKUP_MODEL = "llama3-70b-8192"
 
-# --- Groq API Setup ---
-client = Groq(api_key=api)
+# Streamlit UI Config
+st.set_page_config(
+    page_title="AI Code Generator",
+    page_icon="🚀",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
 
-# Function to get code from StarCoder model
-def generate_code(summary, language):
-    try:
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": f"Generate efficient {language} code for the following task without any explanations or comments: {summary}",
-                }
-            ],
-            model="starcoder",  # Specify the model you want to use
-            stream=False,
-        )
-        generated_code = chat_completion.choices[0].message.content
-        return generated_code
-    except Exception as e:
-        st.error(f"Error generating code: {e}")
-        return ""
-
-# --- Streamlit Interface ---
-st.set_page_config(page_title="Generative AI Code Generator", page_icon="👨‍💻", layout="wide")
+# Custom CSS for Neon Theme
+def inject_custom_css():
+    st.markdown("""
+    <style>
+        .main { background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%); }
+        .stTextInput input { color: #4fffb0 !important; background: rgba(0,0,0,0.3) !important; }
+        .stButton>button { background: rgba(79,255,176,0.2) !important; border: 2px solid #4fffb0 !important; }
+        .code-editor { border: 2px solid #4fffb0; border-radius: 10px; padding: 1rem; background: rgba(0,0,0,0.5); }
+    </style>
+    """, unsafe_allow_html=True)
 
-# Page Title
-st.title("Generative AI Code Generator Using StarCoder")
+inject_custom_css()
 
-# Summary Input
-summary = st.text_area("Enter the Summary of the Task", "For example: Create a function to add two numbers.")
-language = st.selectbox("Select the Programming Language", ["Python", "Java", "JavaScript", "C++"])
+# Initialize Groq Client
+def get_groq_client():
+    GROQ_API_KEY = "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"
+    api_key = GROQ_API_KEY
+    if not api_key:
+        st.error("GROQ_API_KEY not found! Please set it in environment variables or secrets.")
+        return None
+    return Groq(api_key=api_key)
 
-# Generate Code Button
-if st.button("Generate Code"):
-    if summary:
-        # Generate the code using Groq and StarCoder
-        generated_code = generate_code(summary, language)
+# Code Generation Functions
+def generate_code(query, model, client):
+    try:
+        completion = client.chat.completions.create(
+            model=model,
+            messages=[{
+                "role": "user",
+                "content": f"Generate Python code for: {query}. Include comments and error handling."
+            }],
+            temperature=0.6,
+            max_tokens=4096,
+            top_p=0.95
+        )
+        return completion.choices[0].message.content
+    except Exception as e:
+        st.error(f"Error generating code: {str(e)}")
+        return None
 
-        if generated_code:
-            st.subheader(f"Generated {language} Code:")
-            st.code(generated_code, language=language.lower())
+# Main App Interface
+def main():
+    client = get_groq_client()
+    if not client:
+        return
 
-            # Modify or remove sections of code (optional)
-            modified_code = st.text_area("Modify the Code (Optional)", value=generated_code, height=200)
+    st.title("AI Code Generator 🚀")
+    
+    col1, col2 = st.columns([3, 1])
+    with col1:
+        query = st.text_input("Describe your coding requirement:")
+    
+    if 'generated_code' not in st.session_state:
+        st.session_state.generated_code = None
+    
+    if col2.button("🔄 New Code"):
+        st.session_state.generated_code = None
 
-            # Download Code Button (Download as TXT)
-            st.download_button(
-                label="Download Code",
-                data=modified_code or generated_code,  # Use modified_code if edited, otherwise generated_code
-                file_name="generated_code.txt",
-                mime="text/plain"
-            )
+    if query and not st.session_state.generated_code:
+        with st.spinner("Generating code..."):
+            code = generate_code(query, PRIMARY_MODEL, client)
+            if not code:
+                st.warning("Trying backup model...")
+                code = generate_code(query, BACKUP_MODEL, client)
+            
+            if code:
+                st.session_state.generated_code = code
+            else:
+                st.error("Failed to generate code. Please try again.")
 
-# New Code Button
-if st.button("Generate New Code"):
-    summary = ""  # Clear the summary input
-    st.rerun()  # Refresh the page to clear inputs
+    if st.session_state.generated_code:
+        with st.expander("Generated Code", expanded=True):
+            st.markdown(f"```python\n{st.session_state.generated_code}\n```")
 
-# Footer Information
-st.markdown("---")
-st.write("💻 Powered by Streamlit | AI Code Generation by Groq and StarCoder")
+if __name__ == "__main__":
+    main()