Good app

app.py

@@ -10,20 +10,22 @@ def submit_prompt(prompt: str):
     return backend.router(prompt)
 
 def start_app()-> Tuple[bool, bool, bool]:
-    print("Starting app")
     return (
         gr.update(visible=False),  # landing visible
         gr.update(visible=True),   # app visible
         gr.update(visible=False),  # start_button visible
     )
 
-def change_vote(backdown: bool, id: str) -> Tuple[bool]:
-    print(id)
-    response = requests.post(url + "/v2/backdownvote", json={"backdown": backdown, "_id": id})
-    return (gr.update(visible=False),)
+def change_vote( _id:str, backdown: bool,) -> Tuple[bool, bool]:
+    print(backdown, _id)
+    response = requests.post(url + "/v2/backdownvote", json={"backdown": backdown, "_id": _id})
+    return (
+        gr.update(visible=False),
+        gr.update(visible=False)
+    )
 
 def record_vote(prompt: str, left_chat: List, right_chat: List,
-               left_model: str, right_model: str, vote_type: int, moreConsumingn) -> Tuple[str, bool, bool, bool, bool]:
+                left_model: str, right_model: str, energy, moreConsuming, vote_type: int ) -> Tuple[str, bool, bool, bool, bool, bool]:
     """Record a vote for either the left or right model"""
     vote_message = "Is a tie!"
     if vote_type == 0:
@@ -36,7 +38,10 @@ def record_vote(prompt: str, left_chat: List, right_chat: List,
                                                                  "left_model": left_model, "right_model": right_model,
                                                                  })
     changeVisible = False
-    if((moreConsuming == "left" and vote_type == 0) or (moreConsuming == "right" and vote_type == 1)):
+    jsonResponse = response.json()
+    _id = jsonResponse["id"]
+
+    if((moreConsuming == "izquierda" and vote_type == 0) or (moreConsuming == "derecha" and vote_type == 1)):
        changeVisible = True
 
     #result, left_model, buttons[0], buttons[1], tievote_btn, model_names_row,
@@ -46,15 +51,16 @@ def record_vote(prompt: str, left_chat: List, right_chat: List,
         gr.update(interactive=False),               # right_vote_btn interactive
         gr.update(interactive=False),               # tie_btn interactive
         gr.update(visible=True),               # model_names_row visible
-        gr.update(visible=True),               # backdown_row visible
+        gr.update(visible=changeVisible),               # backdown_row visible
+        _id
     )
 
 
-def send_prompt(prompt: str , moreConsuming) -> Tuple[List, List, str, str, bool, bool, bool, bool]:
+def send_prompt(prompt: str) -> Tuple[List, List, str, str, bool, bool, bool, bool, str]:
     response = requests.post(url + "/v2/query", json={"prompt": prompt})
     jsonResponse = response.json()
     print(jsonResponse)
-    if(jsonResponse["status"] == 200 and jsonResponse["message"]):
+    if(jsonResponse["status"] == 200):
         moreConsuming = jsonResponse["message"]["moreConsumption"]
     return (
          [{"role":"assistant", "content": jsonResponse["answers"][0]}], # left_output
@@ -65,6 +71,7 @@ def send_prompt(prompt: str , moreConsuming) -> Tuple[List, List, str, str, bool
          gr.update(interactive=True, visible=True),
          gr.update(interactive=True, visible=True),
          gr.update(visible=False),
+         moreConsuming
     )
 
 # Initialize Gradio Blocks
@@ -138,15 +145,14 @@ with gr.Blocks(css="footer{display:none !important}") as demo:
             value="🤝  It's a Tie!", visible=True, interactive=False
         )
         with gr.Column(visible=False) as backdown_row:
-            backdown_txt = gr.HTML("""<h2>Do you want to change your vote, knowing that the selected model consumes significantly more?</h2>""")
+            backdown_txt = gr.HTML("""<h2> ¿Sabiendo que la respuesta que no has elegido consume menos energía cambiarías tu elección o la mantendrías?</h2>""")
             with gr.Row():
-                no_backdown_btn = gr.Button(value="No", visible=True, interactive=True)
-                backdown_btn = gr.Button(value="Yes", visible=True, interactive=True)
+                no_backdown_btn = gr.Button(value="Mantengo la respuesta", visible=True, interactive=True)
+                backdown_btn = gr.Button(value="Cambiaría de respuesta", visible=True, interactive=True)
         with gr.Row(visible=False) as model_names_row:
             left_model = gr.Textbox(label="Left Model", interactive=False)
             right_model = gr.Textbox(label="Right Model", interactive=False)
 
-
         result = gr.Textbox(label="Result", interactive=False, visible=False)
         with gr.Group():
             with gr.Row():
@@ -162,39 +168,39 @@ with gr.Blocks(css="footer{display:none !important}") as demo:
         tie_count = gr.State(0)
 
         # Define interactions
-        textbox.submit(fn=lambda *args: send_prompt(*args, moreConsuming),
+        textbox.submit(fn=lambda *args: send_prompt(*args),
                 inputs=[textbox],
                 outputs=[chatbot[0], chatbot[1], left_model, right_model,
-                buttons[0], buttons[1], tievote_btn, model_names_row,
+                buttons[0], buttons[1], tievote_btn, model_names_row, moreConsuming
             ])
 
 
         buttons[0].click(
             lambda *args: record_vote(*args, 0),
-            inputs=[textbox, chatbot[0], chatbot[1], left_model, right_model, moreConsuming],
-            outputs=[result,buttons[0], buttons[1], tievote_btn, model_names_row, backdown_row]
+            inputs=[textbox, chatbot[0], chatbot[1], left_model, right_model, gr.State(value=False), moreConsuming],
+            outputs=[result,buttons[0], buttons[1], tievote_btn, model_names_row, backdown_row, _id]
         )
 
         buttons[1].click(
             lambda *args: record_vote(*args, 1),
-            inputs=[textbox, chatbot[0], chatbot[1], left_model, right_model, moreConsuming],
-            outputs=[result,buttons[0], buttons[1], tievote_btn, model_names_row, backdown_row]
+            inputs=[textbox, chatbot[0], chatbot[1], left_model, right_model, gr.State(value=False), moreConsuming],
+            outputs=[result,buttons[0], buttons[1], tievote_btn, model_names_row, backdown_row, _id]
         )
 
         tievote_btn.click(
             lambda *args: record_vote(*args, 2),
-            inputs=[textbox, chatbot[0], chatbot[1], left_model, right_model, gr.State(value=False)],
-            outputs=[result, buttons[0], buttons[1], tievote_btn, model_names_row,]
+            inputs=[textbox, chatbot[0], chatbot[1], left_model, right_model, gr.State(value=False), moreConsuming],
+            outputs=[result,buttons[0], buttons[1], tievote_btn, model_names_row, backdown_row, _id]
         )
         backdown_btn.click(
-            lambda *args: change_vote(*args, _id, True),
-            inputs=[],
+            lambda *args: change_vote(*args, True),
+            inputs=[_id],
             outputs=[backdown_row]
         )
         no_backdown_btn.click(
-            lambda *args: change_vote(*args, _id, False),
-            inputs=[],
-            outputs=[backdown_row]
+            lambda *args: change_vote(*args, False),
+            inputs=[_id],
+            outputs=[backdown_row, model_names_row]
         )
 
         # Project Description
@@ -235,6 +241,5 @@ with gr.Blocks(css="footer{display:none !important}") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=["static"], show_api=False, share=False)
-
+    demo.launch(allowed_paths=["static"], show_api=False, share=True)