Viewing File: /home/ubuntu/combine_ai/writing_aid/gradio_interface.py

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

import gradio as gr
from src.main import *

# Define the function that will be called when the user enters a query
def custom_llm_model(user_id, description, use_case, language, tone, more_details, additional_info):
    # response = KB_main(query,"UI_demo")
    response = data_querying(user_id, description, use_case, language, tone, more_details, additional_info)
    return response


# Gradio interface
iface = gr.Interface(
    fn=custom_llm_model,
    inputs=[
        gr.Textbox(label="user_id"),
        gr.Textbox(label="description"),
        # Write all use cases here
        gr.Radio(label="use_case", choices=["Email Responder", "Rewrite Content", "Autowrite", "Expert Writing Assistant", "Cover Letter", "Post Generator", "SEO Friendly Blog Post", "Blog Post Title Generator", "Caption Generator", "SEO Keyword Generator", "Keywords Extractor", "Marketing Assistant", "Blog Post Generator", "Copy Writing", "Text Summarizer", "Resume Optimizer", "Speech Writer", "Text Expander", "Text Editor", "Text Paraphraser", "Proofreader", "Proposal Generator", "Financial Advisor", "Legal Advisor", "Medical Advisor", "Career Advisor", "Content Moderation", "Explain Like I'm 5", "Academic Essay Writer", "Question Answering"]), 
        gr.Radio(label="language", choices=["English", "Hindi", "Tamil"]),
        gr.Dropdown(label="tone", choices=["Formal", "Informal", "Appreciative", "Assertive", "Casual", "Conversational", "Direct", "Enthusiastic", "Friendly", "Humorous", "Inspirational", "Persuasive", "Professional", "Respectful", "Sincere", "Sympathetic", "Urgent", "Uplifting", "Warm"]),
        gr.Textbox(label="more_details"),
        gr.Textbox(label="additional_info")
    ],
    outputs="text",
    title="Writing Assistant",
    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.201.118", server_port=5020)