Listening preview and some more patches

src/components/ExamEditor/Exercises/Writing/index.tsx

@@ -15,15 +15,13 @@ interface Props {
     exercise: WritingExercise;
 }
 
-const Writing: React.FC<Props> = ({ sectionId }) => {
+const Writing: React.FC<Props> = ({ sectionId, exercise }) => {
     const { currentModule, dispatch } = useExamEditorStore();
     const { edit } = useExamEditorStore((store) => store.modules[currentModule]);
     const { generating, genResult, state } = useExamEditorStore(
         (state) => state.modules[currentModule].sections.find((section) => section.sectionId === sectionId)!
     );
-
-    const exercise = state as WritingExercise;
-
+    
     const [local, setLocal] = useState(exercise);
     const [prompt, setPrompt] = useState(exercise.prompt);
     const [loading, setLoading] = useState(generating && generating == "exercises");
@@ -43,7 +41,9 @@ const Writing: React.FC<Props> = ({ sectionId }) => {
             dispatch({ type: "UPDATE_SECTION_STATE", payload: { sectionId: sectionId, update: newExercise } });
         },
         onDiscard: () => {
+            setEditing(false);
             setLocal(exercise);
+            setPrompt(exercise.prompt);
         },
     });
 
@@ -67,10 +67,6 @@ const Writing: React.FC<Props> = ({ sectionId }) => {
         // eslint-disable-next-line react-hooks/exhaustive-deps
     }, [genResult, dispatch, sectionId, setEditing, currentModule]);
 
-    useEffect(() => {
-        setLocal(state as WritingExercise);
-    }, [state]);
-
     useEffect(() => {
         setEditingAlert(prompt !== local.prompt, setAlerts);
     }, [prompt, local.prompt]);