:tada: initial commit

.gitignore

@@ -0,0 +1,2 @@
+.idea/
+venv/

README.md

@@ -1,10 +1,10 @@
 ---
 title: WebcamGPT
-emoji: 🏢
-colorFrom: red
-colorTo: gray
+emoji: 💬📸
+colorFrom: pink
+colorTo: purple
 sdk: gradio
-sdk_version: 4.1.1
+sdk_version: 3.50.2
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,122 @@
+import base64
+import os
+import uuid
+
+import cv2
+import gradio as gr
+import numpy as np
+import requests
+
+MARKDOWN = """
+# WebcamGPT 💬 + 📸
+
+webcamGPT is a tool that allows you to chat with video using OpenAI Vision API.
+
+Visit [awesome-openai-vision-api-experiments](https://github.com/roboflow/awesome-openai-vision-api-experiments) 
+repository to find more OpenAI Vision API experiments or contribute your own.
+"""
+AVATARS = (
+    "https://media.roboflow.com/spaces/roboflow_raccoon_full.png",
+    "https://media.roboflow.com/spaces/openai-white-logomark.png"
+)
+IMAGE_CACHE_DIRECTORY = "data"
+API_URL = "https://api.openai.com/v1/chat/completions"
+
+
+def preprocess_image(image: np.ndarray) -> np.ndarray:
+    image = np.fliplr(image)
+    return cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+
+
+def encode_image_to_base64(image: np.ndarray) -> str:
+    success, buffer = cv2.imencode('.jpg', image)
+    if not success:
+        raise ValueError("Could not encode image to JPEG format.")
+
+    encoded_image = base64.b64encode(buffer).decode('utf-8')
+    return encoded_image
+
+
+def compose_payload(image: np.ndarray, prompt: str) -> dict:
+    base64_image = encode_image_to_base64(image)
+    return {
+        "model": "gpt-4-vision-preview",
+        "messages": [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": prompt
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": f"data:image/jpeg;base64,{base64_image}"
+                        }
+                    }
+                ]
+            }
+        ],
+        "max_tokens": 300
+    }
+
+
+def compose_headers(api_key: str) -> dict:
+    return {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+
+
+def prompt_image(api_key: str, image: np.ndarray, prompt: str) -> str:
+    headers = compose_headers(api_key=api_key)
+    payload = compose_payload(image=image, prompt=prompt)
+    response = requests.post(url=API_URL, headers=headers, json=payload).json()
+
+    if 'error' in response:
+        raise ValueError(response['error']['message'])
+    return response['choices'][0]['message']['content']
+
+
+def cache_image(image: np.ndarray) -> str:
+    image_filename = f"{uuid.uuid4()}.jpeg"
+    os.makedirs(IMAGE_CACHE_DIRECTORY, exist_ok=True)
+    image_path = os.path.join(IMAGE_CACHE_DIRECTORY, image_filename)
+    cv2.imwrite(image_path, image)
+    return image_path
+
+
+def respond(api_key: str, image: np.ndarray, prompt: str, chat_history):
+    if not api_key:
+        raise ValueError(
+            "API_KEY is not set. "
+            "Please follow the instructions in the README to set it up.")
+
+    image = preprocess_image(image=image)
+    cached_image_path = cache_image(image)
+    response = prompt_image(api_key=api_key, image=image, prompt=prompt)
+    chat_history.append(((cached_image_path,), None))
+    chat_history.append((prompt, response))
+    return "", chat_history
+
+
+with gr.Blocks() as demo:
+    gr.Markdown(MARKDOWN)
+    with gr.Row():
+        webcam = gr.Image(source="webcam", streaming=True)
+        with gr.Column():
+            api_key_textbox = gr.Textbox(
+                label="OpenAI API KEY", type="password")
+            chatbot = gr.Chatbot(
+                height=500, bubble_full_width=False, avatar_images=AVATARS)
+            message_textbox = gr.Textbox()
+            clear_button = gr.ClearButton([message_textbox, chatbot])
+
+    message_textbox.submit(
+        fn=respond,
+        inputs=[api_key_textbox, webcam, message_textbox, chatbot],
+        outputs=[message_textbox, chatbot]
+    )
+
+demo.launch(debug=False, show_error=True)

requirements.txt

@@ -0,0 +1,4 @@
+numpy
+opencv-python
+requests
+gradio==3.50.2