Update app.py

app.py

@@ -3,20 +3,22 @@ import requests
 import os
 import base64
 from PIL import Image
+from io import BytesIO  # Correctly import BytesIO from the io module
 
 # Set page title
 st.set_page_config(page_title="Image Caption Generator", layout="centered")
 
+# API key from environment variable
+API_KEY = os.environ.get("NEBIUS_API_KEY")
+
 # UI for the app
 st.title("Image Caption Generator")
 st.write(
     "Upload an image, and this app will generate a detailed caption for it using the Nebius AI API."
 )
 
-# Sidebar for API key
-api_key = st.sidebar.text_input(
-    "Enter Nebius API Key", type="password", help="Add your Nebius API key here."
-)
+if not API_KEY:
+    st.error("API key not found. Please set the `NEBIUS_API_KEY` environment variable.")
 
 # Function to call Nebius API
 def generate_caption(image_base64, api_key):
@@ -49,17 +51,17 @@ def generate_caption(image_base64, api_key):
 # File uploader for image
 uploaded_image = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
 
-if uploaded_image and api_key:
+if uploaded_image and API_KEY:
     # Convert the image to base64
     image = Image.open(uploaded_image)
-    buffered = st.BytesIO()
+    buffered = BytesIO()  # Use BytesIO from the io module
     image.save(buffered, format="PNG")
     image_base64 = base64.b64encode(buffered.getvalue()).decode()
 
     # Generate caption
     st.image(image, caption="Uploaded Image", use_column_width=True)
     st.write("Generating caption...")
-    result = generate_caption(image_base64, api_key)
+    result = generate_caption(image_base64, API_KEY)
 
     # Display the result
     if "error" in result:
@@ -68,7 +70,9 @@ if uploaded_image and api_key:
         caption = result.get("messages", [{}])[-1].get("content", [{}])[0].get("text", "No caption generated.")
         st.subheader("Generated Caption")
         st.write(caption)
+elif not API_KEY:
+    st.warning("Please set the `NEBIUS_API_KEY` environment variable.")
 else:
-    st.info("Please upload an image and provide your API key.")
+    st.info("Please upload an image.")
 
 st.sidebar.write("Built with ❤️ by OpenAI GPT-4")