Update app.py

app.py

@@ -5,11 +5,12 @@ from openai import OpenAI
 import os
 import base64
 import io
-import glob
 
+# For Gradio spaces, it's better to use secrets for API keys
 client = OpenAI(
-    api_key=os.getenv('API_KEY'),
-    base_url=os.getenv('API_BASE'))
+    api_key=os.environ.get('OPENAI_API_KEY'),
+    base_url=os.environ.get('OPENAI_API_BASE')
+)
 
 MODELS = [
     "google/gemini-flash-1.5",
@@ -46,14 +47,27 @@ def analyze_label(image, model):
     except Exception as e:
         return f"Error analyzing image: {str(e)}"
 
-# Get example images
-example_images = sorted(glob.glob("images (*.png"))
+# Define example images
+example_images = [
+    'images (1).png',
+    'images (2).png',
+    'images (3).png',
+    'images (4).png',
+    'images (5).png',
+    'images (6).png',
+    'images (7).png',
+    'images (8).png',
+    'images (9).png',
+    'images (10).png',
+    'images (11).png',
+    'images (12).png'
+]
 
 iface = gr.Interface(
     fn=analyze_label,
     inputs=[
         gr.Image(type="pil", label="Upload a picture of a product label, food item, or menu"),
-        gr.Dropdown(choices=MODELS, label="Select Model")
+        gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[0])
     ],
     outputs=gr.Textbox(label="Analysis"),
     title="🍏 Sehat GPT 🥦",
@@ -61,5 +75,5 @@ iface = gr.Interface(
     examples=[[img, MODELS[0]] for img in example_images]
 )
 
-# For local testing, uncomment the line below
-# iface.launch(debug=True)
+# For Gradio spaces, we use this to launch the interface
+iface.launch()