Upload 4 files

Pages/pages_File-chat.py

@@ -0,0 +1,74 @@
+import streamlit as st
+import requests
+import utils
+
+# Custom CSS
+with open('styles.css') as f:
+    css = f.read()
+
+st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
+
+## LOGO and TITLE
+## -------------------------------------------------------------------------------------------
+# Show the logo and title side by side
+col1, col2 = st.columns([1, 4])
+with col1:
+    st.image("brainbot.png", width=100)
+with col2:
+    st.title("File-Chat")
+
+question = None
+llm = st.session_state["llm"]
+
+if "current_file" in st.session_state:
+    current_file = st.session_state['current_file']
+    if st.sidebar.button("Upload New File"):
+        st.switch_page("BrainBot.py")
+    st.subheader("Your file has been uploaded successfully. You can now chat with it.")
+    st.success(current_file)
+    question = st.chat_input("Type your question here...")
+
+else:
+    st.warning("Upload a file to begin chat with it.")
+    if st.button("Upload File"):
+        st.switch_page("BrainBot.py")
+
+## CHAT
+# Clear the file chat history if user has uploaded a new file
+if st.session_state['uploaded_file'] == True:
+    st.session_state['file_chat_history'] = []
+
+# Display the file chat history
+for message in st.session_state['file_chat_history']:
+    with st.chat_message("user"):
+        st.write(message["Human"])
+    with st.chat_message("ai"):
+        st.markdown(utils.format_response(message["AI"]))
+
+## QUESTION - WITH CHAT HISTORY
+## -------------------------------------------------------------------------------------------
+# Retrieve the answer to the question asked by the user 
+if question is not None:
+    # Display the question entered by the user in chat
+    with st.chat_message("user"):
+        st.write(question)
+
+    resource = "file"
+
+    try:
+        # Send POST request to a FastAPI endpoint to retrieve an answer for the question
+        data = {"question": question, "resource": resource}
+        FASTAPI_URL = f"http://localhost:8000/answer_with_chat_history/{llm}"
+
+        with st.spinner("Generating response..."):
+            response = requests.post(FASTAPI_URL, json=data)
+            # Append the response to the chat history
+            st.session_state['file_chat_history'].append({"Human": question, "AI": response.text})
+            st.session_state['uploaded_file'] = False
+            # Display the AI's response to the question in chat
+            with st.chat_message("ai"):
+                # Format the response
+                formatted_response = utils.format_response(response.text)
+                st.markdown(formatted_response)
+    except Exception as e:
+                st.switch_page("error.py")

Pages/pages_Image-scan.py

@@ -0,0 +1,67 @@
+import streamlit as st
+import requests
+import utils
+
+# Custom CSS
+with open('styles.css') as f:
+    css = f.read()
+
+st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
+
+## LOGO and TITLE
+## -------------------------------------------------------------------------------------------
+# Show the logo and title side by side
+col1, col2 = st.columns([1, 4])
+with col1:
+    st.image("brainbot.png", width=100)
+with col2:
+    st.title("Image-Scan")
+
+llm = st.session_state["llm"]
+
+if "current_image" in st.session_state:
+    current_image = st.session_state['current_image']
+    if st.sidebar.button("Upload New Image"):
+        st.switch_page("BrainBot.py")
+    st.subheader("Your image has been uploaded successfully.")
+    st.success(current_image)
+else:
+    st.warning("Upload an image to interpret it.")
+    if st.button("Upload Image"):
+        st.switch_page("BrainBot.py")
+
+## CHAT
+# Clear the image chat history if user has uploaded a new image
+if st.session_state['uploaded_image'] == True:
+    st.session_state['image_chat_history'] = []
+
+# Display the image chat history
+for image in st.session_state['image_chat_history']:
+    with st.chat_message("user"):
+        st.image(image["path"], caption=current_image)
+    with st.chat_message("ai"):
+        st.markdown(utils.format_response(image["Description"]))        
+
+## IMAGE
+# Display the image uploaded by the user
+if "temp_img_path" in st.session_state and st.session_state['uploaded_image'] == True:
+    temp_img_path = st.session_state['temp_img_path']
+    with st.chat_message("human"):
+        st.image(temp_img_path, width=300, caption=current_image)
+
+    try:
+        # Send POST request to a FastAPI endpoint with temporary image path
+        FASTAPI_URL = f"http://localhost:8000/image/{llm}"
+        with st.spinner("Interpreting image..."):
+            response = requests.post(FASTAPI_URL, json={"image_path": temp_img_path})
+            # Append the image and response to the chat history
+            st.session_state['image_chat_history'].append({"path": temp_img_path, "Description": response.text})
+            st.session_state['uploaded_image'] = False
+
+            # Display the AI's interpretation of the image in chat
+            with st.chat_message("assistant"):
+                # Format the response
+                formatted_response = utils.format_response(response.text)
+                st.markdown(formatted_response)
+    except Exception as e:
+        st.switch_page("error.py")

Pages/pages_Web-chat.py

@@ -0,0 +1,72 @@
+import streamlit as st
+import requests
+import utils
+
+# Custom CSS
+with open('styles.css') as f:
+    css = f.read()
+
+st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
+
+## LOGO and TITLE
+## -------------------------------------------------------------------------------------------
+# Show the logo and title side by side
+col1, col2 = st.columns([1, 4])
+with col1:
+    st.image("brainbot.png", width=100)
+with col2:
+    st.title("Web-Chat")
+
+question = None
+llm = st.session_state["llm"]
+
+if "current_website" in st.session_state:
+    current_website = st.session_state['current_website']
+    if st.sidebar.button("Upload New Webpage Link"):
+        st.switch_page("BrainBot.py")
+    st.subheader("Your website content has been uploaded successfully. You can now chat with it.")
+    st.success(current_website)
+    question = st.chat_input("Type your question here...")
+else:
+    st.warning("Upload a webpage link to begin chat with it.")
+    if st.button("Upload Webpage Link"):
+        st.switch_page("BrainBot.py")
+
+## CHAT
+# Clear the web chat history if user has uploaded a new webpage link
+if st.session_state['uploaded_link'] == True:
+    st.session_state['web_chat_history'] = []
+
+# Display the web chat history
+for message in st.session_state['web_chat_history']:
+    with st.chat_message("user"):
+        st.write(message["Human"])
+    with st.chat_message("ai"):
+        st.markdown(utils.format_response(message["AI"]))
+
+## QUESTION - WITH CHAT HISTORY
+## -------------------------------------------------------------------------------------------
+# Retrieve the answer to the question asked by the user 
+if question is not None:
+    # Display the question entered by the user in chat
+    with st.chat_message("user"):
+        st.write(question)
+
+    resource = "web"
+
+    try:
+        # Send POST request to a FastAPI endpoint to retrieve an answer for the question
+        data = {"question": question, "resource": resource}
+        FASTAPI_URL = f"http://localhost:8000/answer_with_chat_history/{llm}"
+        with st.spinner("Generating response..."):
+            response = requests.post(FASTAPI_URL, json=data)
+            # Append the response to the chat history
+            st.session_state['web_chat_history'].append({"Human": question, "AI": response.text})
+            st.session_state['uploaded_link'] = False
+            # Display the AI's response to the question in chat
+            with st.chat_message("ai"):
+                # Format the response
+                formatted_response = utils.format_response(response.text)
+                st.markdown(formatted_response)
+    except Exception as e:
+        st.switch_page("error.py")

Pages/pages_error.py

@@ -0,0 +1,18 @@
+import streamlit as st
+
+# Custom CSS
+with open('styles.css') as f:
+    css = f.read()
+
+st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
+
+## LOGO and TITLE
+## -------------------------------------------------------------------------------------------
+# Show the logo and title side by side
+col1, col2 = st.columns([1, 4])
+with col1:
+    st.image("brainbot.png", width=100)
+with col2:
+    st.title("Error")
+
+st.error("Oops - Something went wrong! Please try again.")