update the app to use default API_KEY if it's not provided

app.py

@@ -6,10 +6,9 @@ from PIL import Image
 import numpy as np
 import os
 
-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 write something first."
     
@@ -23,6 +22,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,8 +65,11 @@ def sketch_to_text(image):
 # Create the Gradio interface
 with gr.Blocks() as iface:
     gr.Markdown("# Pix2Latex")
-    gr.Markdown("Transcribing handwritten forumla into latex with Llama3.2 instruct")
-    
+    gr.Markdown("Transcribing handwritten forumla into latex with Llama3.2 instruct. [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):
         input_image = gr.ImageEditor()
 
@@ -72,7 +79,7 @@ with gr.Blocks() as iface:
         with gr.Column(scale=1):
             output2 = gr.Markdown(label="Rendered")
     
-    input_image.change(fn=sketch_to_text, inputs=input_image, outputs=[output1, output2])
+    input_image.change(fn=sketch_to_text, inputs=[input_image, api_key], outputs=[output1, output2])
     
     gr.Markdown("How to use: 1. write your formula in the box above. 2. See it in real time, have fun doing math?")