import gradio as gr
from functionality import get_collection, predict

def read_file(file):
    if file is not None:
        with open(file.name, 'r') as f:
            file_data = f.read()
        return file_data
    else:
        return None

with gr.Blocks() as demo:
    gr.Markdown("# **📝 AI-powered Academic Research Assistant 📝**")
    gr.Markdown("""**AI-powered Academic Research Assistant** is a tool which helps to 
                ensure the *correct grammar* and *academic style* in the scientific papers.

                It also could help with *writing needed parts* or *proposing possible ideas*
                for describing what you want in appropriate way.

                ## 📥 Down bellow you should choose appropriate parameters for your goals and then wait a little for the responce!""")

    gr.Markdown('📨 Write the text you what to expand or upload corresponding text file.')    

    with gr.Tab('Write Text 📖'):
        gr.Markdown("⚙️ *Hint*: to ensure more effective work of 'Fix Academic Style' try to make your sentences not too long (<= 20 words).")
        input_prompt = gr.Textbox(label='Initial Text 📝',
                            placeholder='Write here your research text!',
                            lines=9,)
    with gr.Tab('Upload File 📩'):
        gr.Markdown("⚙️ *Hint*: to ensure more effective work of 'Fix Academic Style' try to make your sentences not too long (<= 20 words).")
        txt_file = gr.File(file_types=['text',], label='Upload Text File',)
        txt_file.change(read_file, inputs=txt_file, outputs=input_prompt)

            
    gr.Markdown('✏️ Fill parameters for your needs')
    with gr.Row(variant='panel', equal_height=True):
        request_goal = gr.Radio(label='🤔 Specify the purpose of your request.',
                                info="Pick one:",
                                choices=['Write Text (Part)', 'Fix Academic Style', 'Fix Grammar', ],
                                value='Write Text (Part)',)
        
        with gr.Accordion("❗️ In case you need to Write Text (Part) choose appropriate option!", open=False):
            part_to_write = gr.CheckboxGroup(label="""📋 What part for Assistant to write? (here you need to 
                                             specify what part of your research you need to complete.)""", 
                                            info="""You may chose as many as needed:""",
                                            choices=['Abstract', 'Introduction', 
                                                     'Methodology', 'Discussion', 'Conclusion', 'Full Text',],
                                            value='Abstract',)
    
    with gr.Row(equal_height=True):
        submit_btn = gr.Button('Confirm! ✅')
        clear_btn = gr.Button('Clear ❌', min_width=611)

    gr.Markdown('##### 📌 Assistant Responce')
    gr.Markdown("In case you did not satisfy with the responce try to paraphrase!")

    responce = gr.Textbox(label="Generated Text 👨🏼‍💻", 
                        info="""You may face some page jumps, it is a bug which will be fixed. Just wait for the completion of text generation. 
                        Sorry for inconvenience(""",
                        lines=9, 
                        placeholder='Here the generated text will appear!', 
                        show_label=True,
                        show_copy_button=True,
                        autofocus=True, 
                        autoscroll=True,)
    
    submit_btn.click(fn=predict, 
                    inputs=[request_goal, part_to_write, input_prompt,], 
                    outputs=[responce], 
                    scroll_to_output=True, queue=True)
    clear_btn.click(lambda: (None, None, 'Write Text (Part)', 'Abstract', None), None, 
                outputs=[input_prompt, txt_file, request_goal, part_to_write, responce])

if __name__ == "__main__":
    get_collection()
    demo.launch()