Update app.py

app.py

@@ -1,22 +1,72 @@
+from googleapiclient.discovery import build
 import streamlit as st
+import requests
+from openai import OpenAI
+import os
 
-# x = st.slider('Select a value')
-# st.write(x, 'squared is', x * x)
+client = OpenAI()
+client.key = os.getenv("OPENAI_API_KEY")
+GOOGLE_API_DEV_KEY = os.getenv("GOOGLE_API_DEV_KEY")
+
+
+def search_amazon(query: str, num: int = 10) -> str:
+    service = build(
+        "customsearch", "v1", developerKey=GOOGLE_API_DEV_KEY
+    )
+
+    res = (
+        service.cse()
+        .list(
+            q=query,
+            cx="103114c8487ce4aa1",
+            num=num,
+        )
+        .execute()
+    )
+
+    links = ""
+
+    for item in res['items']:
+        links += f"- [{item['title']}]({item['link']})\n"
+
+    return links
 
-import streamlit as st
-import requests
 
 # Streamlit interface
-st.title('POST Request Sender')
-url = st.text_input('Enter the URL', 'https://openai.com/completion')
-if st.button('Send POST Request'):
+st.title('Vision Hop')
+url = st.text_input('Enter the image URL', 'Image URL')
+if st.button('Shop'):
     # Make a POST request
     try:
-        response = requests.post(url, data={'sample_data': 'value'})
-        if response.status_code == 200:
-            st.success('POST request successful!')
-            st.json(response.json())
-        else:
-            st.error(f'Request failed with status code {response.status_code}')
+        response = client.chat.completions.create(
+            model="gpt-4-vision-preview",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text",
+                            "text": "Summarize the image as a search string for Amazon.com"},
+#                         "text": "Identify all the items available to be purchased and for each item list the search terms you would to find them on Amazon.com.  Put a number before the search terms"},
+
+                        {
+                            "type": "image_url",
+                            "image_url": {
+                                "url": url,
+                            },
+                        },
+                    ],
+                }
+            ],
+            max_tokens=300,
+        )
+        search_str = response.choices[0].message.content
+        print(search_str)
+        st.write(search_str)
+
+        amazon_output = search_amazon(search_str)
+        print(amazon_output)
+        st.write(amazon_output)
+
     except Exception as e:
         st.error(f'An error occurred: {e}')
+