import gradio as gr
from transformers import pipeline
import requests
import json
import os

decade = pipeline(model="tonyassi/fashion-clothing-decade")

def mistral(decade):
    url = os.environ.get('MISTRAL_URL')
    
    # Define the prompt
    prompt = "write a fun,short description of the " + decade + " in 1-2 sentences"

    # Mistral API call
    payload = json.dumps({
      "key": os.environ.get('MISTRAL_KEY'),
      "messages": [
              {
                  "role": "user",
                  "content": prompt
              },
          ],
      "max_tokens": 1000
    })
    
    headers = {
      'Content-Type': 'application/json'
    }

    # API response
    response = requests.request("POST", url, headers=headers, data=payload)
    response = json.loads(response.text)

    return response['message']

def greet(img):
    # Predict decade from image
    pred = decade(images=img)

    # Write output text
    res = """
# """ + pred[0]['label'] 
    
    return res

iface = gr.Interface(fn=greet,
                     title='Which Decade Are You From?',
                     description="""
by [Tony Assi](https://www.tonyassi.com/)
                     
This space uses the [fashion-clothing-decade](https://huggingface.co/tonyassi/fashion-clothing-decade) image classification model. Please ❤️ this Space.

I build custom AI apps for companies. <a href="mailto: tony.assi.media@gmail.com">Email me</a> for business inquiries.

![](https://cdn.discordapp.com/attachments/1120417968032063538/1184251611388850257/all.png?ex=658b4b42&is=6578d642&hm=6b853c5b1e92d07701496f5fcb3106c6cef15c66a1a238ceedb214b4d9348245&)
""",
                     inputs=gr.Image(type="pil"), 
                     outputs=gr.Markdown(),
                     theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"),
                     examples=[['./examples/1910s.jpg'],['./examples/1920s.jpg'],['./examples/1930s.jpg'],['./examples/1940s.jpg'],['./examples/1950s.jpg'],['./examples/1960s.jpg'],['./examples/1970s.jpg'],['./examples/1980s.jpg'],['./examples/1990s.jpg'],['./examples/2000s.jpg'],]
                     )
iface.launch()