Add loading gif when generating solution and add button for generating Carbon screenshot

app.py

@@ -1,3 +1,5 @@
+import urllib
+
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -135,12 +137,24 @@ def run():
     submit_button = st.button("Solve")
 
     if submit_button:
-        st.text("Solution:")
+        text = st.text("Generating solution...")
+        # gif from https://giphy.com/gifs/alan-DfSXiR60W9MVq
+        gif_runner = st.image("./loading.gif")
         output = generate_solution(
             model, tokenizer, question, starter_code, temperature, num_beams
         )
+        text.empty()
+        gif_runner.empty()
+
+        st.text("Solution:")
         st.code(output, language="python")
 
+        # Create link to carbon to make a nice screenshot of the generated code
+        url_code = urllib.parse.quote(f"# {question}\n{output}")
+        st.markdown(
+            f"[Carbon Screenshot](https://carbon.now.sh/?bg=rgba%280%2C0%2C0%2C0%29&t=seti&wt=none&l=python&ds=false&dsyoff=20px&dsblur=68px&wc=true&wa=false&pv=56px&ph=56px&ln=false&fl=1&fm=Hack&fs=14px&lh=133%25&si=false&es=2x&wm=false&code={url_code})"
+        )
+
 
 if __name__ == "__main__":
     run()