Fixed Image to Model!!!

app.py

@@ -7,9 +7,7 @@ import streamlit as st
 import logging
 from helpers.constant import *
 from helpers.chat import basicChat, guidedMM, mmChat
-import os
-import tempfile
-import requests
+import base64
 
 logger = logging.getLogger(__name__)
 logging.basicConfig(filename='app.log', level=logging.INFO)
@@ -28,13 +26,16 @@ if "systemPrompt" not in st.session_state:
 st.set_page_config(page_title="IMSA Math Helper v0.1")
 st.title("IMSA Math Helper v0.1")
 with st.sidebar:
+    st.info("""This prototype is designed to help students explore the potential and risks of using LLMs to help learn math. 
+    This version (v.01) exists to demonstrate an end-to-end proof of concept for the students to engage with to identify
+    hazards within the system design. Future iterations will include the constraints they devise to improve the system.""")
     # User selects a model
     model_choice = st.radio("Please select the model:", options=["Llama","QWEN"])
     update_model(model_choice)
     logger.info(f"Model changed to {model_choice}.")
     systemPrompt = st.radio("Designate a control persona:",options=["Model","Tutor"])
     st.session_state.systemPrompt = systemPrompt
-    liveExperience = st.checkbox("Check for live experience, currently impoeverished.")
+    liveExperience = st.checkbox("Check to enable the live experience.")
 
 st.subheader(f"This experience is currently running on {st.session_state.model}.")
 
@@ -48,21 +49,17 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 if liveExperience:
-    st.warning("The experience is currently nonfunctional as I attempt to figure out how to expose the image URL to the HF inference API via streamlit...")
-    enable = st.checkbox("Enable camera")
-    picture = st.camera_input("Take a picture of your math work", disabled=not enable)
+    enable = st.checkbox("Enable camera to upload a picture of your math problem.")
+    picture = st.camera_input("Take a picture", disabled=not enable)
 
     if picture is not None:
-        # Save the image to a temporary file
-        temp_dir = tempfile.mkdtemp()
-        temp_image_path = os.path.join(temp_dir, "picture.png")
-        with open(temp_image_path, "wb") as f:
-            f.write(picture.getbuffer())
-        extIP = requests.get("https://ipv4.icanhazip.com").text 
-        img_url = f"https://{extIP}:8501{temp_image_path}"
+        st.image(picture)
+        picBytes= picture.getvalue()
+        b64IMG = base64.b64encode(picBytes).decode('utf-8')
+        img_url = f"data:image/jpeg;base64,{b64IMG}"
         guidedMM(st.session_state.systemPrompt, img_url)
 else:
-    st.info("Please select an example image to understand the basic potential of the app while I try to figure out how to serve images correctly to the completions API.")
+    st.info("Please select an example image to understand the basic potential of the app before using the live app. You can ask the model a question about these images to see how it responds.")
     example = st.selectbox("Select an example image", options=["Example 1", "Example 2"])
     if example == "Example 1":
         img_url = "https://huggingface.co/spaces/butterswords/MM_Math_Helper/resolve/main/tempDir/example1.png"