Merge pull request #18 from doccano/enhancement/adjust-layout

Update layout

doccano_mini/pages/01_Text_Classification.py

@@ -21,19 +21,18 @@
 
 st.header("Test")
 col1, col2 = st.columns([3, 1])
-text = col1.text_area(label="Please enter your text.", value="", height=300)
+inputs = {"input": col1.text_area(label="Please enter your text.", value="", height=300)}
 
 with col2:
     llm = openai_model_form()
 
 with st.expander("See your prompt"):
-    st.markdown(f"```\n{prompt.format(input=text)}\n```")
+    st.markdown(f"```\n{prompt.format(**inputs)}\n```")
 
 if st.button("Predict"):
     chain = LLMChain(llm=llm, prompt=prompt)
-    response = chain.run(text)
-    label = response.split(":")[1]
-    st.text(label)
+    response = chain.run(**inputs)
+    st.text(response)
 
     chain.save("config.yaml")
     display_download_button()

doccano_mini/pages/09_Task_Free.py

@@ -23,11 +23,16 @@
 prompt = make_task_free_prompt(examples)
 prompt = task_instruction_editor(prompt)
 
-inputs = {column: st.text_input(f"Input for {column}:") for column in columns[:-1]}
+st.header("Test")
+col1, col2 = st.columns([3, 1])
+inputs = {column: col1.text_area(label=f"Input for {column}:", value="", height=300) for column in columns[:-1]}
 
-st.markdown(f"Your prompt\n```\n{prompt.format(**inputs)}\n```")
+with col2:
+    llm = openai_model_form()
+
+with st.expander("See your prompt"):
+    st.markdown(f"```\n{prompt.format(**inputs)}\n```")
 
-llm = openai_model_form()
 if st.button("Predict"):
     chain = LLMChain(llm=llm, prompt=prompt)
     response = chain.run(**inputs)