Added argparse for screen vs camera mode

google-gemini · Dec 17, 2024 · dacb16b · dacb16b
1 parent 379912b
commit dacb16b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/gemini-2/live_api_starter.py b/gemini-2/live_api_starter.py
@@ -30,6 +30,12 @@
 import PIL.Image
 import mss
 
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--mode', type=str, default='camera', help='pixels to stream from', choices=['camera', 'screen'])
+args = parser.parse_args()
+
 from google import genai
 
 if sys.version_info < (3, 11, 0):
@@ -45,8 +51,7 @@
 
 MODEL = "models/gemini-2.0-flash-exp"
 
-MODE = "camera"
-#MODE = "screen"
+MODE = args.mode
 
 client = genai.Client(http_options={"api_version": "v1alpha"})