feat: add pre-synopsis

app.py

@@ -70,14 +70,31 @@ elif functionality == "Generation":
         placeholder="Select genres..."
         )
 
-    button_synopsis = st.button('Get synopsis')
+    complete_synopsis = st.toggle('Synopsis completion')
 
-    if button_synopsis:
-        if len(prod_title.split(' ')) > 0:
-            gen_synopsis = generate_synopsis(model_gen, tokenizer_gen, f"{prod_title} | {option_genres} | ")
-            st.text_area('Generated synopsis', value=gen_synopsis, disabled=True)
-        else:
-            st.write('Write a title for the generator to work !')
+    if complete_synopsis:
+
+        pre_synopsis = st.text_area('Type the beginning of your synopsis')
+
+        button_synopsis = st.button('Get synopsis')
+
+        if button_synopsis:
+            if (len(prod_title.split(' ')) > 0) & option_genres:
+                gen_synopsis = generate_synopsis(model_gen, tokenizer_gen, f"{prod_title} | {option_genres} | {pre_synopsis}")
+                st.text_area('Generated synopsis', value=gen_synopsis, disabled=True)
+            else:
+                st.write('Write a title & select a genre for the generator to work !')
+    
+    else:
+
+        button_synopsis = st.button('Get synopsis')
+
+        if button_synopsis:
+            if len(prod_title.split(' ')) > 0:
+                gen_synopsis = generate_synopsis(model_gen, tokenizer_gen, f"{prod_title} | {option_genres} | ")
+                st.text_area('Generated synopsis', value=gen_synopsis, disabled=True)
+            else:
+                st.write('Write a title for the generator to work !')
 
 else:
     st.write("Select a functionality ! 😊")