Update pages/4_TwoImageDescriptor.py

pages/4_TwoImageDescriptor.py

@@ -1,10 +1,6 @@
 import streamlit as st
 import base64
-import anthropic
-
-# Assuming anthropic is a package that provides an Anthropic client for interacting with Claude
-# and it's installed or defined somewhere in your project
-from anthropic import Anthropic
+from anthropic import Anthropic  # Assuming anthropic is a package for interacting with Claude
 
 def image_to_base64(image):
     """Convert the uploaded image to base64."""
@@ -51,19 +47,36 @@ def describe_image(image, api_key):
     return message.content
 
 def main():
-    st.title("Image Description with Claude")
+    st.title("Image Description with Claude 3 Opus")
     
     api_key = st.text_input("Enter your Claude API Key:", type="password")
+
+    st.header("Upload Images")
+    col1, col2 = st.columns(2)
     
-    uploaded_image = st.file_uploader("Drag and drop an image here", type=["jpg", "jpeg", "png"])
+    with col1:
+        view_image = st.file_uploader("Upload View Image", type=["jpg", "jpeg", "png"], key="view")
+    with col2:
+        buffer_image = st.file_uploader("Upload Buffer Image", type=["jpg", "jpeg", "png"], key="buffer")
     
-    if st.button("Describe Image") and uploaded_image is not None:
-        if api_key:
-            description = describe_image(uploaded_image, api_key)
-            st.text("Description from Claude:")
-            st.write(description)
-        else:
+    if st.button("Describe Images"):
+        if not api_key:
             st.error("Please enter a valid API key.")
+            return
+        
+        if view_image is not None:
+            view_description = describe_image(view_image, api_key)
+            st.text("View Image Description from Claude:")
+            st.write(view_description)
+        else:
+            st.error("Please upload a View image.")
+
+        if buffer_image is not None:
+            buffer_description = describe_image(buffer_image, api_key)
+            st.text("Buffer Image Description from Claude:")
+            st.write(buffer_description)
+        else:
+            st.error("Please upload a Buffer image.")
 
 if __name__ == "__main__":
     main()