Add src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,291 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import requests
+import json
+import web3
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+abi = [{"anonymous":False,"inputs":[{"indexed":False,"internalType":"string","name":"ipfsHash","type":"string"}],"name":"Store","type":"event"},{"inputs":[],"name":"getHashes","outputs":[{"internalType":"string[]","name":"","type":"string[]"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"string","name":"ipfsHash","type":"string"}],"name":"storeHash","outputs":[],"stateMutability":"nonpayable","type":"function"}]
+# function that uploads the results to ipfs
+def upload_json_to_ipfs(data):
+    try:
+        url = "https://api.pinata.cloud/pinning/pinJSONToIPFS"
+
+        print("starting upload to ipfs")
+
+        # check for JWT secret
+        if "PinataJWT" not in st.secrets:
+            st.write("No JWT secret found, please add your JWT in a secret titled \"PinataJWT\"")
+            return
+
+        jwt_token = st.secrets["PinataJWT"].strip()
+        headers = {
+            "Authorization": f"Bearer {jwt_token}",
+            "Content-Type": "application/json"
+        }
+
+        # Convert Python dictionary to JSON string
+        response = requests.post(url, headers=headers, json=data)
+
+        if response.status_code == 200:
+            # Print the IPFS hash from the successful response
+            ipfs_hash = response.json().get("IpfsHash")
+            return ipfs_hash
+        else:
+            st.write(f"Failed to upload JSON. Status code: {response.status_code}")
+            st.write(response.text)
+            return None
+    except Exception as e:
+        st.write(f"Error uploading to Pinata: {e}")
+
+# function that uploads to blockchain
+def upload_to_blockchain(hash):
+    print("starting blockchain upload")
+    w3 = web3.Web3(web3.HTTPProvider(st.secrets["infura"]))
+
+    # create an instance of our contract
+    contract = w3.eth.contract(address=st.secrets["ContractAddress"], abi = abi)
+
+    # Call your function: 11155111 is Sepolia's id
+    call_function = contract.functions.storeHash(hash).build_transaction({"chainId": 11155111, 
+                                                                            "from": st.secrets["EthWallet"],
+                                                                            "nonce": w3.eth.get_transaction_count(st.secrets["EthWallet"])})
+
+    # Sign transaction
+    signed_tx = w3.eth.account.sign_transaction(call_function, private_key=st.secrets["pk"])
+
+    # Send transaction
+    send_tx = w3.eth.send_raw_transaction(signed_tx.raw_transaction)
+
+    # Wait for transaction receipt
+    tx_receipt = w3.eth.wait_for_transaction_receipt(send_tx)
+
+    print("ETH Hash:")
+    print(tx_receipt.logs[0].transactionHash.hex())
+
+    return tx_receipt.logs[0].transactionHash.hex()
+
+def get_ipfs_hashes():
+    w3 = web3.Web3(web3.HTTPProvider(st.secrets["infura"]))
+
+    # Create an instance of the contract
+    contract = w3.eth.contract(address=st.secrets["ContractAddress"], abi=abi)
+
+    # Call the getHashes function
+    try:
+        ipfs_hashes = contract.functions.getHashes().call()
+        return ipfs_hashes
+    except Exception as e:
+        st.write(f"Error retrieving hashes: {e}")
+        return []
+    
+def retreive_ipfs_hash_data(hashes):
+    results = []
+    for ipfs_hash in hashes:
+        url = f"https://gateway.pinata.cloud/ipfs/{ipfs_hash}"
+        try:
+            response = requests.get(url)
+            if response.status_code == 200:
+                data = response.json()
+                results.append({"hash": ipfs_hash, "data": data})
+            else:
+                results.append({"hash": ipfs_hash, "error": f"Failed to retrieve data (status {response.status_code})"})
+        except Exception as e:
+            results.append({"hash": ipfs_hash, "error": str(e)})
+    return results
+
+
+# function that handles survey submission
+# sets up ipfs and blockchain
+def submission(survey_data):
+    ipfs_hash = upload_json_to_ipfs(survey_data)
+    if ipfs_hash:
+        print("IPFS Upload Successful")
+        print(ipfs_hash)
+        upload_to_blockchain(ipfs_hash)
+total_number_pages = 4
+placeholder_buttons = None
+
+Q1_radio_options = ["Pizza","Burgers","Pasta","Hot Dogs"]
+Q2_radio_options = ["Water","Soda","Coffee","Tea"]
+Q3_radio_options = ["Yes","No"]
+Q4_radio_options = ["N/A", "Strongly Disagree", "Disagree", "Neutral", "Agree", "Strongly Agree"]
+
+
+# Function that records radio element changes
+def radio_change(element, state, key):
+   st.session_state[state] = element.index(st.session_state[key]) # Setting previously selected option
+
+def multi_change(element, state, key):
+   st.session_state[state] = []
+   for selected_option in st.session_state[key]:
+       st.session_state[state].append(selected_option)
+
+# Function that disables the last button while data is uploaded to IPFS
+def button_disable():
+   st.session_state['disabled'] = True
+
+def answer_change(state, key):
+   st.session_state[state] = st.session_state[key]
+
+st.set_page_config(page_title='IPFS-Based Survey',)
+st.title('Survey Test')
+
+st.markdown("<style>.row-widget.stButton {text-align: center;}</style>", unsafe_allow_html=True)
+st.markdown("<style>.big-font {font-size:24px;}</style>", unsafe_allow_html=True)
+
+
+if "current_page" not in st.session_state:
+    st.session_state["current_page"] = 1
+    st.session_state["Q1"] = None
+    st.session_state["Q2"] = None
+    st.session_state["Q3"] = None
+    st.session_state["Q4"] = None
+    st.session_state["Q5"] = None
+    st.session_state["disabled"] = False
+
+# Page 1; Video
+if st.session_state["current_page"]  == 1:
+
+    st.markdown("""<p class="big-font">This is a description for my survey</p>""", unsafe_allow_html=True)
+
+    st.radio(label     = "What is your favorite food?", 
+             options   = Q1_radio_options, 
+             index     = None if st.session_state["Q1"] == None else st.session_state["Q1"], 
+             key       = 'Q1_radio', 
+             on_change = radio_change, 
+             args      = (Q1_radio_options, "Q1", "Q1_radio",))
+
+    st.markdown("""<style> div[class*="stRadio"] > label > div[data-testid="stMarkdownContainer"] > p {font-size: 18px;}</style> <br><br>""", unsafe_allow_html=True)
+
+
+    st.multiselect(label     = "Select all of the drinks that you like.", 
+             default   = None if st.session_state["Q2"] == None else st.session_state["Q2"], 
+             options   = Q2_radio_options, 
+             key       = 'Q2_multi', 
+             on_change = multi_change, 
+             args      = (Q2_radio_options, "Q2", "Q2_multi",))
+
+    st.markdown("""<style> div[class*="stMulti"] > label > div[data-testid="stMarkdownContainer"] > p {font-size: 18px;}</style> <br><br>""", unsafe_allow_html=True)
+
+
+    placeholder = st.empty()
+
+    if st.button('Next', key='next_button_page_1'):
+        all_answered = True
+        if st.session_state["Q1"] == None or st.session_state["Q1"] == []:
+            all_answered = False
+        if st.session_state["Q2"] == None or st.session_state["Q2"] == []:
+            all_answered = False
+        if all_answered:
+            st.session_state["current_page"] += 1
+            st.rerun()
+        else:
+            with placeholder.container():
+                st.warning("Please answer all the questions on this page.", icon="⚠️")
+
+    st.progress(st.session_state["current_page"]/total_number_pages, text="Progress")
+
+
+elif st.session_state["current_page"] == 2:
+
+    st.video("https://www.youtube.com/watch?v=dQw4w9WgXcQ") 
+    st.markdown("""<p style='font-size:18px;'>Get rick rolled!</p>""", unsafe_allow_html=True)
+    st.selectbox(label     = "Was that a good joke?", 
+             options   = Q3_radio_options, 
+             index     = None if st.session_state["Q3"] == None else st.session_state["Q3"], 
+             key       = 'Q3_radio', 
+             on_change = radio_change, 
+             args      = (Q3_radio_options, "Q3", "Q3_radio",))
+
+    st.markdown("""<style> div[class*="stSelectbox"] > label > div[data-testid="stMarkdownContainer"] > p {font-size: 18px;}</style> <br><br>""", unsafe_allow_html=True)
+
+
+    placeholder = st.empty()
+
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button('Back'):
+            st.session_state["current_page"] -= 1
+            st.rerun()
+    with col2:
+        if st.button('Next'):
+            all_answered = True
+            if st.session_state["Q3"] == None or st.session_state["Q3"] == []:
+                all_answered = False
+            if all_answered:
+                st.session_state["current_page"] += 1
+                st.rerun()
+            else:
+                with placeholder.container():
+                    st.warning("Please answer all the questions on this page.", icon="⚠️")
+
+    st.progress(st.session_state["current_page"]/total_number_pages, text="Progress")
+
+
+elif st.session_state["current_page"] == 3:
+
+    st.radio(label     = "Miami is the best school.", # Likert
+             options   = Q4_radio_options, 
+             index     = None if st.session_state["Q4"] == None else st.session_state["Q4"], 
+             key       = 'Q4_radio', 
+             on_change = radio_change, 
+             args      = (Q4_radio_options, "Q4", "Q4_radio",))
+
+    st.markdown("""<style> div[class*="stRadio"] > label > div[data-testid="stMarkdownContainer"] > p {font-size: 18px;}</style> <br><br>""", unsafe_allow_html=True)
+
+
+    if st.session_state["Q5"] == None:
+        st.session_state["Q5"] = 50
+    st.slider(label="How happy are you today?",min_value=0,max_value=100, 
+             value= st.session_state["Q5"], 
+             key = "Q5_slider",
+             on_change = answer_change, 
+             args = ("Q5", "Q5_slider",))
+    st.markdown("""<style> div[class*="stSlider"] > label > div[data-testid="stMarkdownContainer"] > p {font-size: 18px;}</style> <br><br>""", unsafe_allow_html=True)
+
+
+    placeholder = st.empty()
+
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button('Back'):
+            st.session_state["current_page"] -= 1
+            st.rerun()
+    with col2:
+        if st.button('Next'):
+            all_answered = True
+            if st.session_state["Q4"] == None or st.session_state["Q4"] == []:
+                all_answered = False
+            if st.session_state["Q5"] == None or st.session_state["Q5"] == []:
+                all_answered = False
+            if all_answered:
+                st.session_state["current_page"] += 1
+                st.rerun()
+            else:
+                with placeholder.container():
+                    st.warning("Please answer all the questions on this page.", icon="⚠️")
+
+    st.progress(st.session_state["current_page"]/total_number_pages, text="Progress")
+
+
+elif st.session_state["current_page"] == 4: # Last Page
+    st.markdown('<p class="big-font">Thank you for participating! <br> Click on the button below to submit your answers. </p>', unsafe_allow_html=True)
+    st.button('Submit Responses', disabled = st.session_state["disabled"], on_click = button_disable)
+    if st.session_state["disabled"]:
+        with st.spinner(r"$\textsf{\normalsize Storing data on IPFS and Ethereum. This operation might take a few minutes. Please wait to receive your confirmation code!}$"):
+            try:
+                response = {
+                    "Q1": Q1_radio_options[st.session_state["Q1"]],
+                    "Q2": st.session_state["Q2"],
+                    "Q3": st.session_state["Q3"],
+                    "Q4": st.session_state["Q4"],
+                    "Q5": st.session_state["Q5"],
+                }
+                submission(response)
+            except Exception as e:
+                print(e)
+                st.error(f'An error ocurred. Here is the error message: {e}', icon="🚨")
+
+    if st.button('Back'):
+        st.session_state["current_page"] -= 1
+        st.rerun()
+    st.progress(st.session_state["current_page"]/total_number_pages, text="Progress")