app refactored

app.py

@@ -164,116 +164,116 @@
 #  *Educational Purpose Only* | 🌱 **SDG Guardians AI - 2024** |  *For a greener East Africa*
 # """)
 
-import streamlit as st
-from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
+# import streamlit as st
+# from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
 
-# Page configuration
-st.set_page_config(page_title="Climate Chatbot - Uganda", layout="wide")
+# # Page configuration
+# st.set_page_config(page_title="Climate Chatbot - Uganda", layout="wide")
 
-# Custom CSS for shadow effect
-st.markdown(
-    """
-    <style>
-    .stChatInput {
-        box-shadow: 0px 10px 20px rgba(0, 0, 0, 0.4);  /* Strong shadow */
-        border-radius: 10px;
-        padding: 12px;
-        background: white;
-    }
-    .stChatInput::before {
-        content: "";
-        position: absolute;
-        width: 100%;
-        height: 15px;
-        left: 0;
-        background: linear-gradient(to top, rgba(0, 0, 0, 0.3), rgba(0, 0, 0, 0)); /* Fading effect */
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
-# Load model
-@st.cache_resource
-def load_climate_bert():
-    tokenizer = AutoTokenizer.from_pretrained("NinaErlacher/ClimateBERTqa")
-    model = AutoModelForQuestionAnswering.from_pretrained("NinaErlacher/ClimateBERTqa")
-    return pipeline("question-answering", model=model, tokenizer=tokenizer)
-
-qa_pipeline = load_climate_bert()
-
-# Function to check if question is climate-related
-def is_climate_related(question):
-    climate_keywords = ["climate", "global warming", "deforestation", "carbon", "sustainability", 
-                        "renewable", "pollution", "green energy", "climate action", "afforestation"]
-    return any(keyword in question.lower() for keyword in climate_keywords)
-
-# Function to check if Uganda is mentioned
-def is_uganda_related(question):
-    return "uganda" in question.lower() or "east africa" in question.lower()
-
-# Function to generate response
-def generate_response(user_question, context):
-    if not is_climate_related(user_question):
-        return "I'm here to discuss climate change. Try asking about Uganda's climate, sustainability, or environmental issues."
-    if not is_uganda_related(user_question):
-        return "This chatbot focuses on climate change in Uganda. Try asking about Uganda's environmental challenges."
+# # Custom CSS for shadow effect
+# st.markdown(
+#     """
+#     <style>
+#     .stChatInput {
+#         box-shadow: 0px 10px 20px rgba(0, 0, 0, 0.4);  /* Strong shadow */
+#         border-radius: 10px;
+#         padding: 12px;
+#         background: white;
+#     }
+#     .stChatInput::before {
+#         content: "";
+#         position: absolute;
+#         width: 100%;
+#         height: 15px;
+#         left: 0;
+#         background: linear-gradient(to top, rgba(0, 0, 0, 0.3), rgba(0, 0, 0, 0)); /* Fading effect */
+#     }
+#     </style>
+#     """,
+#     unsafe_allow_html=True
+# )
+
+# # Load model
+# @st.cache_resource
+# def load_climate_bert():
+#     tokenizer = AutoTokenizer.from_pretrained("NinaErlacher/ClimateBERTqa")
+#     model = AutoModelForQuestionAnswering.from_pretrained("NinaErlacher/ClimateBERTqa")
+#     return pipeline("question-answering", model=model, tokenizer=tokenizer)
+
+# qa_pipeline = load_climate_bert()
+
+# # Function to check if question is climate-related
+# def is_climate_related(question):
+#     climate_keywords = ["climate", "global warming", "deforestation", "carbon", "sustainability", 
+#                         "renewable", "pollution", "green energy", "climate action", "afforestation"]
+#     return any(keyword in question.lower() for keyword in climate_keywords)
+
+# # Function to check if Uganda is mentioned
+# def is_uganda_related(question):
+#     return "uganda" in question.lower() or "east africa" in question.lower()
+
+# # Function to generate response
+# def generate_response(user_question, context):
+#     if not is_climate_related(user_question):
+#         return "I'm here to discuss climate change. Try asking about Uganda's climate, sustainability, or environmental issues."
+#     if not is_uganda_related(user_question):
+#         return "This chatbot focuses on climate change in Uganda. Try asking about Uganda's environmental challenges."
     
-    result = qa_pipeline(question=user_question, context=context)
-    return result['answer']
+#     result = qa_pipeline(question=user_question, context=context)
+#     return result['answer']
 
-# Session state for chat history
-if "history" not in st.session_state:
-    st.session_state.history = []
+# # Session state for chat history
+# if "history" not in st.session_state:
+#     st.session_state.history = []
 
-# Sidebar - Chat History & Clear Button
-with st.sidebar:
-    st.title("Chat History")
-    for idx, (question, answer) in enumerate(st.session_state.history[::-1]):
-        with st.expander(f"💬 {question}"):
-            st.write(f"**Chatbot:** {answer}")
-    st.markdown("---")
-    if st.button("🗑️ Clear Chat History"):
-        st.session_state.history = []
-        st.rerun()
-    st.info("🌱 *Ask about climate change in Uganda.*")
+# # Sidebar - Chat History & Clear Button
+# with st.sidebar:
+#     st.title("Chat History")
+#     for idx, (question, answer) in enumerate(st.session_state.history[::-1]):
+#         with st.expander(f"💬 {question}"):
+#             st.write(f"**Chatbot:** {answer}")
+#     st.markdown("---")
+#     if st.button("🗑️ Clear Chat History"):
+#         st.session_state.history = []
+#         st.rerun()
+#     st.info("🌱 *Ask about climate change in Uganda.*")
+
+# # Main UI
+# st.title("Climate Change Chatbot")
+# st.subheader("Explore climate action and sustainability in Uganda")
+
+# # Sample questions section
+# with st.expander("Need ideas? (Click to expand)"):
+#     st.markdown("""
+#     - **How is Uganda affected by climate change?**
+#     - **What are sustainable farming methods?**
+#     - **How can I reduce my energy use?**
+#     - **What are the risks of deforestation?**
+#     - **Why is tree planting important?**
+#     - **How can youth take action?**
+#     """)
+
+# # Chat container with avatars
+# chat_container = st.container()
+# with chat_container:
+#     for question, answer in st.session_state.history:
+#         with st.chat_message("user"):
+#             st.write(question)
+#         with st.chat_message("assistant"):
+#             st.write(answer)
 
-# Main UI
-st.title("Climate Change Chatbot")
-st.subheader("Explore climate action and sustainability in Uganda")
-
-# Sample questions section
-with st.expander("Need ideas? (Click to expand)"):
-    st.markdown("""
-    - **How is Uganda affected by climate change?**
-    - **What are sustainable farming methods?**
-    - **How can I reduce my energy use?**
-    - **What are the risks of deforestation?**
-    - **Why is tree planting important?**
-    - **How can youth take action?**
-    """)
-
-# Chat container with avatars
-chat_container = st.container()
-with chat_container:
-    for question, answer in st.session_state.history:
-        with st.chat_message("user"):
-            st.write(question)
-        with st.chat_message("assistant"):
-            st.write(answer)
-
-# User input field with shadow effect
-user_input = st.chat_input("Ask about climate change in Uganda...")
-if user_input:
-    context = """
-    Climate change is affecting Uganda and East Africa in various ways, including unpredictable rainfall, 
-    rising temperatures, and prolonged droughts. Sustainable farming, afforestation, and renewable energy 
-    adoption are key solutions to mitigate these effects.
-    """  # Placeholder context
+# # User input field with shadow effect
+# user_input = st.chat_input("Ask about climate change in Uganda...")
+# if user_input:
+#     context = """
+#     Climate change is affecting Uganda and East Africa in various ways, including unpredictable rainfall, 
+#     rising temperatures, and prolonged droughts. Sustainable farming, afforestation, and renewable energy 
+#     adoption are key solutions to mitigate these effects.
+#     """  # Placeholder context
     
-    response = generate_response(user_input, context)
-    st.session_state.history.append((user_input, response))
-    st.rerun()
+#     response = generate_response(user_input, context)
+#     st.session_state.history.append((user_input, response))
+#     st.rerun()
 
 
 # seems to be overcrowding the page, so we can remove it for now.
@@ -303,3 +303,64 @@ if user_input:
 #     """,
 #     unsafe_allow_html=True
 # )
+
+import streamlit as st
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+
+# Page configuration
+st.set_page_config(page_title="ClimateGPT Chatbot - Uganda", layout="wide")
+
+# Load ClimateGPT model
+@st.cache_resource
+def load_climate_gpt():
+    model_name = "eci-io/climategpt-7b"
+    
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
+
+    return pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=512)
+
+climate_gpt_pipeline = load_climate_gpt()
+
+# Function to generate response using ClimateGPT
+def generate_response(user_question):
+    prompt = f"""
+    <|im_start|>system
+    You are ClimateGPT, an expert in climate change. Provide accurate and fact-based responses about Uganda’s climate issues.<|im_end|>
+    <|im_start|>user
+    {user_question}<|im_end|>
+    <|im_start|>assistant
+    """
+    
+    response = climate_gpt_pipeline(prompt)[0]["generated_text"]
+    
+    # Extract only the assistant's response
+    response = response.split("<|im_start|>assistant")[-1].strip()
+    return response
+
+# Chat history
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+# Sidebar - Chat History & Clear Button
+with st.sidebar:
+    st.title("Chat History")
+    for idx, (question, answer) in enumerate(st.session_state.history[::-1]):
+        with st.expander(f"💬 {question}"):
+            st.write(f"**Chatbot:** {answer}")
+    
+    if st.button("🗑️ Clear Chat History"):
+        st.session_state.history = []
+        st.rerun()
+
+# Main UI
+st.title("🌍 ClimateGPT - Uganda")
+st.subheader("Ask about climate change, sustainability, and environmental action in Uganda!")
+
+# User input field
+user_input = st.chat_input("Ask me anything about Uganda's climate...")
+if user_input:
+    response = generate_response(user_input)
+    st.session_state.history.append((user_input, response))
+    st.rerun()