Fix explanations

html_items.py

@@ -60,7 +60,7 @@ def create_poi_html(poi: dict, user_id: str) -> str:
     """
     html += f"""        </tbody>
                     </table>
-                    <div style="margin: 15px 0; padding: 10px; background-color: #3a3a3a; color: white; border-left: 5px solid #7FB77E; border-radius: 5px;" --explanation_type="{poi["explanation_type"]}">
+                    <div style="margin: 15px 0; padding: 10px; background-color: #3a3a3a; color: white; border-left: 5px solid #7FB77E; border-radius: 5px;" --explanation_type="{poi.get("explanation_type", -1)}">
                         <h3>Spiegazione della raccomandazione: ti suggeriamo questo luogo perché ...</h3>
                         <p>{poi["explanation"]}</p>
                     </div>

main.py

@@ -361,13 +361,19 @@ def create_main_app(self):
 
                 # ADVANCED SURVEY
                 with gr.Column(visible=False) as advanced_survey_block:
-                    gr.Markdown("## Valuta questa spiegazione della raccomandazione")
+                    if args.survey_language == "it":
+                        gr.Markdown("## Valuta questa spiegazione della raccomandazione")
+                    else:
+                        gr.Markdown("## Rate this explanation of the recommendation")
 
                     # Stato per tenere traccia del gruppo corrente
                     advanced_current_group_idx = gr.State(0)
 
                     # Carica i dati del questionario
-                    from utils.survey_utils import LIKERT_OPTIONS, SURVEY_STATEMENTS
+                    from utils.survey_utils import SURVEY_OPTIONS
+
+                    SURVEY_STATEMENTS = SURVEY_OPTIONS[args.survey_language].get("statements", {})
+                    LIKERT_OPTIONS = SURVEY_OPTIONS[args.survey_language].get("likert_options", [])
 
                     # Contenitori per i gruppi di affermazioni
                     advanced_containers = []
@@ -379,7 +385,10 @@ def create_main_app(self):
                     max_statements_per_group = 2  # Per i primi due gruppi
                     for i in range(3):
                         with gr.Column(visible=(i == 0)) as container:
-                            gr.Markdown(f"### Parte {i + 1} di 3")
+                            if args.survey_language == "it":
+                                gr.Markdown(f"### Parte {i + 1} di 3")
+                            else:
+                                gr.Markdown(f"### Part {i + 1} of 3")
 
                             group_responses = []
                             # Creiamo spazio per il massimo numero di affermazioni possibili
@@ -815,6 +824,11 @@ if __name__ == "__main__":
         default=5,
         help="Number of top recommendations to show",
     )
+    parser.add_argument(
+        "--survey_language",
+        choices=["it", "en"],
+        help="Language for the survey questions",
+    )
     parser.add_argument(
         "--port", type=int, default=7860, help="Port to run the Gradio app on"
     )

resources/survey_statements.json

@@ -1,18 +1,38 @@
 {
-  "statements": [
-    "Mi ha aiutato a capire se mi piacerà quel posto.",
-    "Mi ha fatto decidere più velocemente se mi piace quel posto.",
-    "Mi ha fatto venire voglia di andare lì.",
-    "Mi è piaciuto come scelgono il posto da consigliare.",
-    "Mi ha fatto capire se i suggerimenti rispecchiano i miei gusti.",
-    "Mi ha fatto capire perché mi consigliano quel posto.",
-    "Mi ha fatto fidare del suggerimento."
-  ],
-  "likert_options": [
-    "In disaccordo",
-    "Parzialmente in disaccordo",
-    "Neutrale",
-    "Parzialmente d'accordo",
-    "D'accordo"
-  ]
+  "it": {
+    "statements": [
+      "Mi ha aiutato a capire se mi piacerà quel posto.",
+      "Mi ha fatto decidere più velocemente se mi piace quel posto.",
+      "Mi ha fatto venire voglia di andare lì.",
+      "Mi è piaciuto come scelgono il posto da consigliare.",
+      "Mi ha fatto capire se i suggerimenti rispecchiano i miei gusti.",
+      "Mi ha fatto capire perché mi consigliano quel posto.",
+      "Mi ha fatto fidare del suggerimento."
+    ],
+    "likert_options": [
+      "In disaccordo",
+      "Parzialmente in disaccordo",
+      "Neutrale",
+      "Parzialmente d'accordo",
+      "D'accordo"
+    ]
+  },
+  "en": {
+    "statements": [
+      "It helped me understand whether I will like that place.",
+      "It made me decide more quickly whether I like that place.",
+      "It made me want to go there.",
+      "I liked how they choose the place to recommend.",
+      "It helped me understand whether the suggestions match my tastes.",
+      "It helped me understand why they recommend that place to me.",
+      "It made me trust the suggestion."
+    ],
+    "likert_options": [
+      "Disagree",
+      "Somewhat disagree",
+      "Neutral",
+      "Somewhat agree",
+      "Agree"
+    ]
+  }
 }

utils/recommender_poi_utils.py

@@ -43,8 +43,9 @@ def display_recommendations(user_id: str, k: int = 10) -> list[dict]:
             random.shuffle(available_pois)
             recommendations = available_pois[:k]
 
-        if LOADED_MOCK_EXPLANATIONS:
-            for rec in recommendations:
+    if LOADED_MOCK_EXPLANATIONS:
+        for rec in recommendations:
+            if "explanation" not in rec:
                 explanation = random.choice(LOADED_MOCK_EXPLANATIONS)
                 rec["explanation"] = explanation.get("explanation", "")
                 rec["explanation_type"] = explanation.get("explanation_type", -1)
@@ -114,7 +115,7 @@ def view_poi_details(
     html = create_poi_html(poi, user_id)
 
     # Ottieni il tipo di spiegazione dal POI
-    explanation_type = poi.get("explanation_type", 0)
+    explanation_type = poi.get("explanation_type", -1)
 
     # Salva le informazioni del POI corrente per il questionario
     poi_info = {

utils/survey_utils.py

@@ -7,9 +7,7 @@ import gradio as gr
 
 RESOURCE_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "resources")
 with open(os.path.join(RESOURCE_DIR, "survey_statements.json"), "r") as f:
-    survey_data = json.load(f)
-SURVEY_STATEMENTS = survey_data.get("statements", [])
-LIKERT_OPTIONS = survey_data.get("likert_options", [])
+    SURVEY_OPTIONS = json.load(f)
 
 
 def feedback_message():