Adding format to answer.

app.py

@@ -49,20 +49,11 @@ try:
         st.markdown(multi)
 
         try:    
-            st.button("Example", type="primary")
             if st.button('Ask for answer'):
                 st.write('The answer is:')            
                 answer = pipeline.give_the_best_answer(mi_dict)
-                st.write(answer)
-            else:
-                mi_dict= {
-                    "prompt":"What is color?",
-                    "A": "Is a name.",
-                    "B":"Is something horrible.",
-                    "C":"I don't know.",
-                    "D":"You should ask someone else.",
-                    "E":"Ask in a pyshic book."
-                }
+                st.write(f"The correct answer is {answer}: {mi_dict[answer]}")
+                st.balloons()
 
         except Exception as e:
             st.error("Something bad happend while trying to infer the answer or with the buttons.")