| import streamlit as st |
| from huggingface_hub import InferenceClient |
|
|
| st.set_page_config(page_title="JEE AI Mentor", page_icon="🎓", layout="centered") |
|
|
| |
| st.markdown(""" |
| <style> |
| #MainMenu {visibility: hidden;} |
| footer {visibility: hidden;} |
| header {visibility: hidden;} |
| |
| .stApp { |
| background-color: #131314; |
| color: #e3e3e3; |
| font-family: 'Segoe UI', Roboto, Helvetica, Arial, sans-serif; |
| } |
| |
| .main-title { |
| text-align: center; |
| font-size: 2rem; |
| font-weight: 500; |
| background: linear-gradient(45deg, #4285f4, #9b51e0); |
| -webkit-background-clip: text; |
| -webkit-text-fill-color: transparent; |
| margin-bottom: 0.2rem; |
| } |
| .main-subtitle { |
| text-align: center; |
| font-size: 0.9rem; |
| color: #80868b; |
| margin-bottom: 1.5rem; |
| } |
| |
| /* Minimalist Rounded Controls */ |
| div[data-baseweb="select"] > div { |
| border-radius: 24px !important; |
| background-color: #1e1f20 !important; |
| border: 1px solid #303134 !important; |
| color: #e3e3e3 !important; |
| padding: 2px 8px !important; |
| } |
| div[data-baseweb="select"] span { color: #e3e3e3 !important; } |
| div[data-testid="stMarkdownContainer"] > p { color: #c4c7c5; font-size: 0.95rem; } |
| |
| /* Message Bubbles Styling */ |
| div[data-testid="stChatMessage"] { |
| background-color: transparent !important; |
| border: none !important; |
| padding: 0.5rem 0 !important; |
| } |
| div[data-testid="stChatMessage"]:has(div[data-testid="chatAvatar-user"]) { text-align: right; } |
| div[data-testid="stChatMessage"]:has(div[data-testid="chatAvatar-user"]) div[data-testid="stMarkdownContainer"] p { |
| display: inline-block; |
| background-color: #2b2a33; |
| padding: 10px 18px; |
| border-radius: 20px 20px 4px 20px; |
| color: #e3e3e3; |
| text-align: left; |
| max-width: 85%; |
| } |
| div[data-testid="stChatMessage"]:has(div[data-testid="chatAvatar-assistant"]) div[data-testid="stMarkdownContainer"] { |
| color: #e3e3e3; font-size: 1.05rem; line-height: 1.6; max-width: 95%; |
| } |
| div[data-testid="chatAvatar-user"], div[data-testid="chatAvatar-assistant"] { display: none !important; } |
| |
| /* Chat Input Layout Configuration */ |
| div[data-testid="stChatInput"] { |
| position: fixed; |
| bottom: 24px; |
| left: 50%; |
| transform: translateX(-50%); |
| width: 90% !important; |
| max-width: 700px; |
| z-index: 99; |
| } |
| div[data-testid="stChatInput"] textarea { |
| border-radius: 28px !important; |
| background-color: #1e1f20 !important; |
| border: 1px solid #303134 !important; |
| color: #e3e3e3 !important; |
| padding: 14px 20px !important; |
| } |
| .block-container { padding-bottom: 180px !important; } |
| </style> |
| """, unsafe_allow_html=True) |
|
|
| |
| st.markdown('<div class="main-title">JEE Focus AI</div>', unsafe_allow_html=True) |
| st.markdown('<div class="main-subtitle">Your personalized Advanced preparation partner</div>', unsafe_allow_html=True) |
|
|
| |
| col1, col2 = st.columns(2) |
| with col1: |
| target_class = st.selectbox("Class context", ["Class 12 Aspirant", "Dropper Batch", "Class 11"], label_visibility="collapsed") |
| with col2: |
| subject = st.selectbox("Subject focus", ["Physics", "Chemistry", "Mathematics"], label_visibility="collapsed") |
|
|
| query_type = st.radio( |
| "Select operational lens:", |
| ["📚 Doubt", "⏳ Backlog", "🔥 Motivation", "🧠 Revision"], |
| horizontal=True, |
| label_visibility="collapsed" |
| ) |
|
|
| |
| if "messages" not in st.session_state: |
| st.session_state.messages = [ |
| {"role": "assistant", "content": "Context established. Adjust constraints above whenever you switch tasks. What problem framework or chapter strategy are we evaluating?"} |
| ] |
|
|
| |
| for message in st.session_state.messages: |
| display_content = message["content"] |
| if "[CONTEXT:" in display_content and "]" in display_content: |
| display_content = display_content.split("]")[-1].strip() |
| |
| with st.chat_message(message["role"]): |
| st.markdown(display_content) |
|
|
| |
| if user_prompt := st.chat_input("Ask me anything..."): |
| |
| context_injector = ( |
| f"[CONTEXT: The student is dealing with a {query_type} query for {subject} relevant to {target_class} status. " |
| "Structure your response specifically through this operational lens.] " |
| ) |
| |
| with st.chat_message("user"): |
| st.markdown(user_prompt) |
| |
| st.session_state.messages.append({"role": "user", "content": context_injector + user_prompt}) |
|
|
| with st.chat_message("assistant"): |
| response_placeholder = st.empty() |
| full_response = "" |
| |
| try: |
| client = InferenceClient() |
| |
| system_instruction = ( |
| "You are an elite, veteran JEE Advanced mentor who coached top 100 rankers. " |
| "You understand the exact pressures of Class 11/12/Dropper PCM, standard coaching modules, and PYQ weights.\n" |
| "CRITICAL: Tailor your response perfectly based on the hidden [CONTEXT] config data:\n" |
| "- If target is 'Class 12 Aspirant': Balance your tips around board management, school practicals, and running syllabus timelines.\n" |
| "- If target is 'Dropper Batch': Assume zero school restrictions, focus heavily on maximizing high-yield execution, rigorous test analysis, and handling drop-year mental pressure.\n" |
| "- If target is 'Class 11': Prioritize fundamental building blocks, tackling initial mechanics/organic shock, and long-term consistency.\n\n" |
| "Further segment by action lens:\n" |
| "- If 'Doubt': Break down the core mathematical or conceptual physics/chem mechanisms step-by-step.\n" |
| "- If 'Backlog': Give a high-yield micro-schedule prioritizing mandatory chapters before deep diving into sub-topics.\n" |
| "- If 'Motivation': Be blunt, realistic, highly encouraging, and cut through decision paralysis.\n" |
| "- If 'Revision': Detail how to make high-yield formula sheets and short notes trackers.\n" |
| "Keep responses concise, bolding critical terms, completely optimized for clean conversational mobile reading. Avoid generic greetings." |
| ) |
| |
| formatted_messages = [{"role": "system", "content": system_instruction}] |
| for msg in st.session_state.messages: |
| formatted_messages.append({"role": msg["role"], "content": msg["content"]}) |
| |
| |
| stream = client.chat.completions.create( |
| model="meta-llama/Meta-Llama-3-8B-Instruct", |
| messages=formatted_messages, |
| max_tokens=800, |
| stream=True |
| ) |
| |
| for chunk in stream: |
| if chunk.choices[0].delta.content: |
| full_response += chunk.choices[0].delta.content |
| response_placeholder.markdown(full_response + "▌") |
| |
| response_placeholder.markdown(full_response) |
| |
| except Exception as e: |
| st.error(f"Transmission bottleneck: {e}") |
| full_response = "Hit a brief connection bump. Mind dropping that question in one more time?" |
| response_placeholder.markdown(full_response) |
|
|
| st.session_state.messages.append({"role": "assistant", "content": full_response}) |
|
|
| |
| st.markdown(""" |
| <div style="text-align: center; font-size: 0.75rem; color: #80868b; font-family: sans-serif; margin-top: 20px;"> |
| Follow Me <a href="https://instagram.com/kosmos.cpp" style="color: #4285f4; text-decoration: none; font-weight: 500;" target="_blank">ig@kosmos.cpp</a> |
| </div> |
| """, unsafe_allow_html=True) |
|
|
