Fix: Move set_page_config to top to avoid StreamlitAPIException

app/app.py

@@ -1,8 +1,11 @@
-
 import streamlit as st
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import torch
 
+#  MUST be the first Streamlit command
+st.set_page_config(page_title="CodeMentor AI", page_icon="💻", layout="centered")
+
+# Load model and tokenizer
 @st.cache_resource
 def load_model():
     model = AutoModelForSeq2SeqLM.from_pretrained("Tuathe/codementor-flan")
@@ -13,11 +16,17 @@ model, tokenizer = load_model()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
-st.set_page_config(page_title="CodeMentor AI", page_icon="💻", layout="centered")
-
-st.markdown("<h1 style='text-align: center;'>CodeMentor AI</h1>", unsafe_allow_html=True)
-st.markdown("<p style='text-align: center; font-size:18px;'>Your AI Coding Interview Assistant</p>", unsafe_allow_html=True)
-
+# Streamlit app UI
+st.markdown(
+    "<h1 style='text-align: center;'>CodeMentor AI</h1>", 
+    unsafe_allow_html=True
+)
+st.markdown(
+    "<p style='text-align: center; font-size:18px;'>Your AI Coding Interview Assistant</p>",
+    unsafe_allow_html=True
+)
+
+# Sidebar info
 with st.sidebar:
     st.title("About CodeMentor AI")
     st.info(
@@ -27,6 +36,7 @@ with st.sidebar:
     st.markdown("---")
     st.markdown("Created by Chetan")
 
+# Chat interface
 user_input = st.text_area("Ask your coding question here:", height=150)
 
 if st.button("Get Answer"):