fix reload

app.py

@@ -106,12 +106,17 @@ elif model_name in ["GPT-2 Small finetuned on Indonesian academic journals"]:
 session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
 
 ALL_PROMPTS = list(PROMPT_LIST[prompt_group_name].keys())+["Custom"]
-session_state.prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
+prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
+
+if session_state.prompt is not None and (prompt != session_state.prompt):
+    session_state.prompt_box = None
+    session_state.text = None
 
 if session_state.prompt == "Custom":
     session_state.prompt_box = "Enter your text here"
 else:
-    session_state.prompt_box = random.choice(PROMPT_LIST[prompt_group_name][session_state.prompt])
+    if session_state.prompt_box is None:
+        session_state.prompt_box = random.choice(PROMPT_LIST[prompt_group_name][session_state.prompt])
 
 session_state.text = st.text_area("Enter text", session_state.prompt_box)
 
@@ -120,6 +125,7 @@ if st.button("Run"):
         lang_predictions, lang_probability = ft_model.predict(session_state.text.replace("\n", " "), k=3)
         if "__label__id" in lang_predictions:
             lang = "id"
+            text = session_state.text
         else:
             lang = lang_predictions[0].replace("__label__", "")
             text = translate(session_state.text, "id", lang)
@@ -158,3 +164,8 @@ if st.button("Run"):
         else:
             # display cat image if no image found
             st.image(image_cat, width=400)
+
+        # Reset state
+        session_state.prompt = None
+        session_state.prompt_box = None
+        session_state.text = None