Update src/interface/ui.py

src/interface/ui.py

@@ -1 +1,76 @@
-# Placeholder for Streamlit interface
+import os
+import base64
+import streamlit as st
+from langchain_core.messages import HumanMessage, AIMessage
+
+def get_base64_image(image_path):
+    with open(image_path, 'rb') as img_file:
+        return base64.b64encode(img_file.read()).decode()
+
+def create_interface(call_model):
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+
+    # Background
+    background_image_path = os.path.abspath("assets/uwf-page-edited.jpg")
+    if os.path.exists(background_image_path):
+        base64_background_image = get_base64_image(background_image_path)
+        st.markdown(
+            f"""
+            <style>
+            .stApp {{
+                background-image: url("data:image/jpeg;base64,{base64_background_image}");
+                background-size: cover;
+                background-position: center;
+                background-repeat: no-repeat;
+                background-attachment: fixed;
+            }}
+            </style>
+            """,
+            unsafe_allow_html=True
+        )
+
+    # Header
+    col1, col2 = st.columns([1, 3])
+    with col1:
+        image_path = os.path.abspath("assets/argie-argonaut-gif-edited.png")
+        if os.path.exists(image_path):
+            st.image(image_path, width=200)
+    with col2:
+        st.markdown("<h1 style='color:black;'>ARGObot: UWF's Custom Q&A Chatbot</h1>", unsafe_allow_html=True)
+
+    # Sample questions
+    st.markdown("### Example questions:")
+    st.markdown("- UWF's values")
+    st.markdown("- Student Rights and Responsibilities")
+    st.markdown("- Student Resources")
+
+    # Display conversation
+    for message in st.session_state.chat_history:
+        if isinstance(message, HumanMessage):
+            st.markdown(f"<div style='background-color:#8DC8E8; padding:10px; border-radius:10px;'><strong>You:</strong> {message.content}</div>", unsafe_allow_html=True)
+        elif isinstance(message, AIMessage):
+            st.markdown(f"<div style='background-color:#E7E7E7; padding:10px; border-radius:10px;'><strong>Chatbot:</strong> {message.content}</div>", unsafe_allow_html=True)
+
+    # Input form
+    with st.form(key="chat_form", clear_on_submit=True):
+        user_input = st.text_input("Enter your question here:")
+        submitted = st.form_submit_button("Submit")
+
+        if submitted:
+            if user_input.strip():
+                state = {
+                    "input": user_input,
+                    "chat_history": st.session_state.chat_history,
+                    "context": "",
+                    "answer": ""
+                }
+                result = call_model(state)
+                if not result.get("answer"):
+                    result["answer"] = "Sorry, I couldn't generate an answer."
+
+                st.session_state.chat_history.append(HumanMessage(user_input))
+                st.session_state.chat_history.append(AIMessage(result["answer"]))
+                st.rerun()
+            else:
+                st.warning("Please enter a question.")