import gradio as gr
import random
import json
import string

def ai(input):
    # Create a list of dictionaries
    dictList = []
    
    # Read from the json file
    with open('data.json', 'r') as openfile:
        for jsonObj in openfile:
            dict = json.loads(jsonObj)
            dictList.append(dict)
    
    
    # Create the dict of keywords
    keywords = dictList[0]
    keys = list(keywords.keys())

    input = gr.Textbox(placeholder = "HI")
    output = gr.Textbox()
    # Prompt the user for a response
    user = input
    user = user.lower()
    user = user.translate(str.maketrans('', '', string.punctuation))
    
    keyword_found = False
    # Check if user input contains a keyword
    for i in range(len(keys)):
        if(keys[i].lower() == user):
            return keywords[keys[i]]
            keyword_found = True
        elif(keys[i].lower() in user):
            return keywords[keys[i]]
            keyword_found = True
            break
    if(keyword_found == False):
        return "Sorry, I do not have a response for that input"



iface = gr.Interface(fn=ai, inputs="text", outputs="text")
iface.launch()