saramneena/AI_Conversational_Data_Science_Tutor

1import os2import streamlit as st3from langchain.memory import ConversationBufferMemory4from langchain.chains import ConversationChain5from langchain_google_genai import ChatGoogleGenerativeAI6 7GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")8 9st.set_page_config(page_title="Conversational AI Data Science Tutor", page_icon="🤖")10 11st.title("🤖 Conversational AI Data Science Tutor")12st.write("Ask me any **Data Science** related question!")13 14if not GEMINI_API_KEY:15    st.error("❌ GEMINI_API_KEY not found. Please add it in Hugging Face → Settings → Variables and secrets.")16else:17    llm = ChatGoogleGenerativeAI(18        model="gemini-1.5-pro",19        google_api_key=GEMINI_API_KEY20    )21    memory = ConversationBufferMemory()22    conversation = ConversationChain(23        llm=llm,24        memory=memory,25        verbose=False26    )27 28    29    if "messages" not in st.session_state:30        st.session_state.messages = []31 32    33    for msg in st.session_state.messages:34        st.chat_message(msg["role"]).markdown(msg["content"])35 36 37    if prompt := st.chat_input("Ask a data science question..."):38        st.session_state.messages.append({"role": "user", "content": prompt})39        st.chat_message("user").markdown(prompt)40        response = conversation.predict(input=prompt)41        st.session_state.messages.append({"role": "assistant", "content": response})42        st.chat_message("assistant").markdown(response)43