from flask import Flask
from gradio_client import Client
import time
from multiprocessing import Process
from transformers import Trainer

trainer = Trainer()
app = Flask(__name__)

@app.route('/')
def hello():
    return '{"status": {"details": {"code": 200, "message": "working"}}}'
@app.route('/api/<string:query>')
def api(query):
    client = Client("https://theblueberry-ai-mist-chat.hf.space/")
    result = client.predict(
		query,	# str  in 'Message' Textbox component
		api_name = "/chat"
	)

    return result

def record_loop():
   while True:
        trainer.log('Loop Running')
        print("loop running")
        time.sleep(5)


if __name__ == "__main__":
   trainer.log('App Starting...')
   p = Process(target=record_loop)
   p.start()  
   app.run(debug=True, use_reloader=False)
   p.join()