fix: form

app.py

@@ -121,6 +121,7 @@ def init_session_state():
     st.session_state["max_new_tokens"] = 512
     st.session_state["repetition_penalty"] = 1.1
     st.session_state["system_prompt"] = "You are a medical expert that provides answers for a medically trained audience"
+    st.session_state["prompt"] = ""
 
 def get_genders():
     return ['Male', 'Female']
@@ -219,6 +220,8 @@ def display_llm_output():
     submitted = form.form_submit_button('Submit')
 
     if submitted:
+        st.session_state["system_prompt"] = system_prompt
+        st.session_state["prompt"] = prompt
         formatted_prompt = format_prompt(prompt_format, system_prompt, prompt)
         print(f"Formatted prompt: {format_prompt}")
         llm_response = get_llm_response(