Update app.py

app.py

@@ -5,7 +5,7 @@ from openai import OpenAI
 # Initialize the Nebius OpenAI client
 client = OpenAI(
     base_url="https://api.studio.nebius.ai/v1/",
-    api_key=os.getenv("OPENAI_API_KEY"),  # Use the environment variable
+    api_key=os.getenv("NEBIUS_API_KEY"),  # Use the environment variable
 )
 
 # Function to process input and return the response
@@ -23,9 +23,12 @@ def analyze_image(image_url):
             max_tokens=300,
         )
         
-        # Extract and return the AI's response
-        return response.choices[0].get("message", {}).get("content", "No response found.")
-    
+        # Extract the AI response
+        if response.choices and "message" in response.choices[0]:
+            return response.choices[0]["message"]["content"]
+        else:
+            return "No valid response received from the API."
+
     except Exception as e:
         return f"Error: {str(e)}"