Update app.py

app.py

@@ -37,6 +37,16 @@ with st.spinner(text="Please wait while the model is loading...."):
     tokenizer = load_tokenizer('nafisehNik/girt-t5-base')
 
 
+def create_instruction(name, about, title, labels, assignees, headline_type, headline, summary):
+    val_list = [name, about, title, labels, assignees, headline_type, headline]
+
+    val_list = ['<|MASK|>' if not element else element for element in val_list]
+    if not summary:
+        summary = '<|EMPTY|>'
+    
+    instruction = f'name: {value_list[0]}\nabout: {value_list[1]}\ntitle: {value_list[2]}\nlabels: {value_list[3]}\nassignees: {value_list[4]}\nheadlines_type: {value_list[5]}\nheadlines: {value_list[6]}\nsummary: {summary}'
+    return instruction
+
 def compute(sample, top_p, top_k, do_sample, max_length, min_length):
 
     inputs = tokenizer(sample, return_tensors="pt").to('cpu')
@@ -90,19 +100,24 @@ with st.sidebar:
         in_name = st.text_input("Name Metadata: ", placeholder="e.g., Bug Report or Feqture Request or Question", on_change=None)
         in_about = st.text_input("About Metadata: ", placeholder="e.g., File a bug report", on_change=None)
 
-        
+        empty_title = st.checkbox('without title')
         if empty_title == False:
             in_title = st.text_input("Title Metadata: ", placeholder="e.g., [Bug]: ", on_change=None)
-            empty_title = st.checkbox('without title')
+        else:
+            in_title = '<|EMPTY|>'    
         
         empty_labels = st.checkbox('without labels')
         if empty_labels == False:
             in_labels = st.text_input("Labels Metadata: ", placeholder="e.g., feature, enhancement", on_change=None)
-        
+        else:
+            in_labels = '<|EMPTY|>'
+
         empty_assignees = st.checkbox('without Assignees')
         if empty_assignees == False:
             in_assignees = st.text_input("Assignees Metadata: ", placeholder="e.g., USER_1, USER_2", on_change=None)
-
+        else:
+            in_assignees = '<|EMPTY|>'
+            
         # if no headlines is selected, force the headlines to be empty as well.
         in_headline_type = st.selectbox(
         'How would you like to be your Headlines?',
@@ -110,6 +125,7 @@ with st.sidebar:
 
         if in_headline_type!='No headlines':
             in_headlines = st.text_area("Headlines: ", placeholder="Enter each headline in one line. e.g.,\nWelcome\nConcise Description\nAdditional Info", on_change=None, height=200)
+            in_headlines = in_headlines.split('\n').strip()
         else:
             in_headline_type = '<|EMPTY|>'
             in_headlines = '<|EMPTY|>'
@@ -127,6 +143,14 @@ with st.sidebar:
         top_p = st.slider("top_p", 0.0, 1.0, 0.92)
         top_k = st.slider("top_k", 0, 100, 0)
 
+    
+    clicked = st.button("Submit", key='prompt')
+
+    with st.spinner("Please Wait..."):
+        prompt = create_instruction(in_name, in_about, in_title, in_labels, in_assignees, in_headline_type, in_headlines, in_summary)
+
+        res = compute(prompt, top_p, top_k, do_sample=True, max_length, min_length)
+        st.code(res, language="python")
 
 tab1, tab2 = st.tabs(["Design GitHub Issue Template", "Manual Prompt"])