gradio- simplify

app.py

@@ -5,17 +5,12 @@ DEMO_SERVER = "4.208.9.167:80"
 
 def run_query( server, policy, prompt):
     if prompt == None or server == None or policy == None:
-        return("⛔ Error: please select an option for stages 1-3")
+        return("⛔ Error: please select an option for stages 1 and 2")
     if len(prompt) == 0 or len(policy) == 0 or len(server) == 0:
         return("⛔ Error: please select an option for stages 1-3")
-    if server != "Authentic confidential VM server":
+    if server != "Authentic and verified confidential VM server":
         return ("⛔ Error: you can only connect to an application running on a Confidential VM")
-    if policy == "Expected Santacoder app policy file":
-        POLICY = "./cce_policy.txt" 
-    elif policy == "Unexpected Hello World app policy file":
-        POLICY = "./hello_world.txt"
-    else:
-        POLICY = "fake.txt"
+    POLICY = "./cce_policy.txt"
     try:
         with SecureSession(f"http://{DEMO_SERVER}", POLICY) as secure_session:
             res = secure_session.post(endpoint="/generate", json={"input_text": prompt})
@@ -30,25 +25,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("<p>You can use this demo to send a function definition to BigCode's open-source Santacoder model and get back an auto-completed function.</p>")
     gr.Markdown("<p>The model is deployed within a highly-isolated Trusted Execution Environment, meaning that we, as the service provider, have no access to the data sent to this model!</p>")
     gr.Markdown("<p>You can see how we deployed the model by checking out the integration section of our <a href='https://blindbox.mithrilsecurity.io/en/latest/docs/how-to-guides/santacoder/'>documentation!</p>")
-    gr.Markdown("><h3>Step 1: Check we are connecting to an authentic confidential VM")
-    gr.Markdown("<p>This first option allows you to choose whether to connect to the Whisper application deployed with BlindBox on a confidential VM or the same application deployed on a dummy server which is not within a confidential VM!<br>This demonstrates how BlindBox blocks requests to non-authentic confidential VMs!</p>")
+    gr.Markdown("><h3>Step 1: Check that we are connecting to an authentic confidential VM")
+    gr.Markdown("<p>This first option allows you to choose whether to connect to the Santacoder application deployed with BlindBox on a verified confidential VM or the same application deployed on a dummy server which is not within a confidential VM!<br>This demonstrates how BlindBox blocks requests to non-authentic confidential VMs!</p>")
     with gr.Column():
         server = gr.Radio(
-        ["Authentic confidential VM server", "Unauthentic dummy server"], label="Select the server you want to connect to"
-        )
-    gr.Markdown("><h3>Step 2: Check we are connecting to the latest official SaaS application image</h3>")
-    gr.Markdown("<p>You can think of this second verification a bit like a checksum. Here, you can see what happens if the end user queries the application whilst providing the latest policy file for our Santacoder application, the latest policy file for a different Hello World application or sends no policy file at all.<br></p>")
-    with gr.Column():
-        policy = gr.Radio(
-        ["Expected Santacoder app policy file", "Unexpected Hello World app policy file", "No policy file"], label="Select your CCE policy file"
+        ["Authentic and verified confidential VM server", "Unauthentic dummy server"], label="Select the server you want to connect to"
         )
-    gr.Markdown("><h3>Step 3: Select your prompt</h3>")
+    gr.Markdown("><h3>Step 2: Select your prompt</h3>")
     gr.Markdown("<p>Select between the following prompt examples we provide.</p>")
     with gr.Column():
         prompt = gr.Radio(
         ["def sum(x, y):", "def print_name(name):", "def hello_world():", "def square_root(nbr):"], label="Select your user prompt"
         )
-    gr.Markdown("><h3>Query the Santacoder model</h3>")
+    gr.Markdown("><h3>Step 3: Query the Santacoder model</h3>")
     with gr.Column():
         trigger = gr.Button("Test query")
     with gr.Column():