Update app.py
Browse files
app.py
CHANGED
|
@@ -5,10 +5,10 @@ from dotenv import load_dotenv
|
|
| 5 |
|
| 6 |
# Load GROQ API key
|
| 7 |
load_dotenv()
|
| 8 |
-
GROQ_API_KEY =
|
| 9 |
|
| 10 |
# Init client
|
| 11 |
-
client =
|
| 12 |
|
| 13 |
# App UI Setup
|
| 14 |
st.set_page_config(page_title="🧠 Code Snippet Generator", page_icon="💡", layout="wide")
|
|
@@ -60,8 +60,8 @@ Input: {prompt}
|
|
| 60 |
messages=[
|
| 61 |
{"role": "user", "content": full_prompt}
|
| 62 |
],
|
| 63 |
-
model="llama-3
|
| 64 |
)
|
| 65 |
output = response.choices[0].message.content
|
| 66 |
st.markdown("### 🧩 Generated Code + Dry Run:")
|
| 67 |
-
st.code(output, language="python")
|
|
|
|
| 5 |
|
| 6 |
# Load GROQ API key
|
| 7 |
load_dotenv()
|
| 8 |
+
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
| 9 |
|
| 10 |
# Init client
|
| 11 |
+
client = Groq(api_key=GROQ_API_KEY)
|
| 12 |
|
| 13 |
# App UI Setup
|
| 14 |
st.set_page_config(page_title="🧠 Code Snippet Generator", page_icon="💡", layout="wide")
|
|
|
|
| 60 |
messages=[
|
| 61 |
{"role": "user", "content": full_prompt}
|
| 62 |
],
|
| 63 |
+
model="llama-3-70b-8192",
|
| 64 |
)
|
| 65 |
output = response.choices[0].message.content
|
| 66 |
st.markdown("### 🧩 Generated Code + Dry Run:")
|
| 67 |
+
st.code(output, language="python")
|