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"),  # Replace with your API key if not using environment variables
)

# Function to interact with the Nebius OpenAI API
def analyze_image(image_path):
    try:
        # Upload the image to a hosting service or convert its path to an accessible URL
        # For simplicity, assume the user provides a valid image URL here
        image_url = image_path  # Replace this with a real image URL if needed

        # API request
        response = client.chat.completions.create(
            model="Qwen/Qwen2-VL-72B-Instruct",  # Ensure this model name is correct
            messages=[
                {
                    "role": "user",
                    "content": [
                        {"type": "text", "text": "What’s in this image?"},
                        {
                            "type": "image_url",
                            "image_url": {"url": image_url},  # Use the proper field for image_url
                        },
                    ],
                }
            ],
            max_tokens=300,
        )

        # Extract the AI's 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)}"


# Gradio interface for uploading an image
with gr.Blocks() as app:
    gr.Markdown("# Image Analysis with Nebius OpenAI")
    with gr.Row():
        image_url_input = gr.Textbox(
            label="Image URL",
            placeholder="Enter a valid image URL for analysis",
        )
        output_text = gr.Textbox(label="AI Response")
    
    analyze_button = gr.Button("Analyze Image")
    analyze_button.click(analyze_image, inputs=image_url_input, outputs=output_text)

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