update app to allow default API key

app.py

@@ -5,11 +5,9 @@ import requests
 import io
 from PIL import Image
 import numpy as np
-
-API_KEY = os.environ['API_KEY']
 URL = os.environ['URL']
 
-def sketch_to_text(image):
+def sketch_to_text(image, api_key):
     if image is None or not isinstance(image, dict) or 'composite' not in image:
         return "Please draw something first."
     
@@ -23,6 +21,11 @@ def sketch_to_text(image):
     pil_image.save(buffered, format="PNG")
     img_str = base64.b64encode(buffered.getvalue()).decode()
 
+    if api_key:
+        API_KEY = api_key
+    else:
+        API_KEY = os.environ['API_KEY']
+        
     # Prepare the API request
     headers = {
         "Content-Type": "application/json",
@@ -61,15 +64,16 @@ def sketch_to_text(image):
 # Create the Gradio interface
 with gr.Blocks() as iface:
     gr.Markdown("# Pictionary with Llama3.2 Instruct")
-    gr.Markdown("Draw something and let Llama guess it!")
-    
+    gr.Markdown("Draw something and let Llama3.2 guess it! [Powered by SambaNova Cloud, Get Your API Key Here](https://cloud.sambanova.ai/apis)")
+    with gr.Row():
+        api_key = gr.Textbox(label="API Key", type="password", placeholder="(Optional) Enter your API key here for more availability")
     with gr.Column(scale=1):
         output = gr.Textbox(label="Description", lines=5)
     
     with gr.Column(scale=1):
         input_image = gr.ImageEditor()
     
-    input_image.change(fn=sketch_to_text, inputs=input_image, outputs=output)
+    input_image.change(fn=sketch_to_text, inputs=[input_image, api_key], outputs=output)
     
     gr.Markdown("How to use: 1. Draw your sketch in the box above. 2. See guessing in real time. Have fun sketching!")