import os
import gradio as gr
from openai import OpenAI

# Initialize the Nebius OpenAI client
client = OpenAI(
    base_url="https://api.studio.nebius.ai/v1/",
    api_key=os.environ.get("NEBIUS_API_KEY"),
)

# Function to process input and return the response
def analyze_image(image_url):
    response = client.chat.completions.create(
        model="Qwen/Qwen2-VL-72B-Instruct",
        messages=[
            {
                "role": "user",
                "content": [
                    {"type": "text", "text": "What’s in this image?"},
                    {
                        "type": "image_url",
                        "image_url": {"url": image_url},
                    },
                ],
            }
        ],
        max_tokens=300,
    )
    # Extract and return the AI's response
    return response.choices[0].get("message", {}).get("content", "No response found.")

# Create the Gradio interface
with gr.Blocks() as app:
    gr.Markdown("# Image Analysis with Nebius OpenAI")
    image_url_input = gr.Textbox(
        label="Image URL",
        placeholder="Enter an image URL for analysis",
    )
    output = gr.Textbox(
        label="AI Response",
        placeholder="The description of the image will appear here.",
    )
    submit_button = gr.Button("Analyze Image")
    submit_button.click(analyze_image, inputs=image_url_input, outputs=output)

# Launch the app
if __name__ == "__main__":
    app.launch()