Update pages/11_Gemini-Cyclic.py

pages/11_Gemini-Cyclic.py

@@ -2,6 +2,7 @@ import streamlit as st
 import uuid
 import google.generativeai as genai
 from io import BytesIO
+from PIL import Image
 
 # Define a dictionary of system prompts mapped to numbers
 system_prompts = {
@@ -12,6 +13,14 @@ system_prompts = {
     '5': "AI Fitness Coach Prompt: Offer fitness and health advice.",
 }
 
+placeholder_images = {
+    '1': Image.open('data/image1.jpg'),
+    '2': Image.open('data/image2.jpg'),
+    '3': Image.open('data/image3.jpg'),
+    '4': Image.open('data/image4.jpg'),
+    '5': Image.open('data/image5.jpg')
+}
+
 # Set your API key
 api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
 
@@ -65,6 +74,9 @@ def process_input_sequence():
     input_sequence = st.session_state.input_sequence
     for digit in input_sequence:
         system_prompt = system_prompts.get(digit, "Default system prompt")
+        # Display the corresponding image for the current digit
+        if digit in placeholder_images:
+            st.image(placeholder_images[digit], caption=f"Image for Prompt {digit}")
         send_message(system_prompt)
 
 # Send message function with system prompt
@@ -104,4 +116,4 @@ input_sequence = st.text_input("Enter your input sequence (e.g., 12345):", value
 send_button = st.button("Send", on_click=process_input_sequence)
 
 # Clear conversation button
-clear_button = st.button("Clear Conversation", on_click=clear_conversation)
+clear_button = st.button("Clear Conversation", on_click=clear_conversation)