Viewing File: /home/ubuntu/codegamaai-test/broker_bot/gradio_interface.py

# Description: This file is used to create a chat interface with Gradio UI.
# =========================================================================

import gradio as gr
from src.knowledge import *
from main import main_text

# Define the function that will be called when the user enters a query
def custom_llm_model(query, phone_number):
    uuid = phone_number
    # response = KB_main(query,"UI_demo")
    response  = main_text(query, uuid)
    return response

# Gradio interface
iface = gr.Interface(
    fn=custom_llm_model,
    inputs=[
        gr.Textbox(label="query"),
        gr.Textbox(label="phone number")
    ],
    outputs="text",
    title="Broker Bot",
    description="please enter your query here",
    allow_flagging=True,  # Disable flagging for inappropriate content
    live=False  # Disable the live updates for this example
)
iface.launch(share=True,server_name="164.52.213.13", server_port=5006)