import json
import requests
import gradio as gr
import random

API_TOKEN = os.getenv("API_TOKEN")
headers = {"Authorization": f"Bearer {API_TOKEN}"}
API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"


prompt="""What would this look like in 2050? Give 5 optimally describing bigrams.
Input:cake shop
Output:Sustainable ingredients,Virtual reality experience,Customizable flavor options,3D printed designs,Robotic assistants
#####
Input:consulting firm
Output:Virtual meetings,AI integration,Sustainability focus,Virtual reality simulations,Blockchain technology
#####
Input:biscuit brand
Output:Sustainable packaging,Plant-based ingredients,Digital marketing,Virtual reality,Smart packaging
#####
"""
  

def query(text):
    #global prompt
    print(text)
    text = prompt+"\nInput:"+text + "\nOutput:"
    #print(text)
    payload =     {
    "inputs": text,
    "parameters": {
      "max_length": 250,
      "temperature": 0.9,
        "seed":random.randint(0, 100),
      },
    }

    data = json.dumps(payload)
    response = requests.request("POST", API_URL, headers=headers, data=data)
    #print('Response here\n',response.content)
    generated_text=json.loads(response.content.decode("utf-8"))[0]['generated_text']
    result = generated_text.replace(text,'').strip()
    result = result.replace("Output:","")
    parts = result.split("###")
    topic = parts[0].strip()
    topic=topic.split("Input:")[0]
    print(topic)
    return topic


with gr.Blocks() as demo:
    gr.Markdown("<h1><center>Ideas of the Future</center></h1>")
    gr.Markdown(
        """ChatGPT based Insights from Decodem.ai for businesses.\nWhile ChatGPT has multiple use cases we have evolved specific use cases/ templates for businesses \n\n This template provides ideas on how a business would look like in the future. Enter a business area and get the results."""
        )
    textbox = gr.Textbox(placeholder="Enter business type here...", lines=1,label='Yourbusiness area')
    btn = gr.Button("Generate")    
    output1 = gr.Textbox(lines=2,label='The future')

    btn.click(query,inputs=[textbox], outputs=[output1])
    examples = gr.Examples(examples=['icecream parlor','space travel','book shop','ecommerce','grocery delivery'],
                           inputs=[textbox])
    

demo.launch()