update: Enhance reaction saving functionality by adding user type selection and updating dataset path

app.py

@@ -12,22 +12,22 @@ def load_data(database_file):
         chunk_embeddings[idx] = t.tensor(df.loc[df.index[idx], "chunk_embeddings"])
     return df, chunk_embeddings
 
-def save_reactions_to_dataset(query, results):
+def save_reactions_to_dataset(user_type, query, results):
     data = {
+        "user_type": [],
         "query": [],
         "retrieved_text": [],
-        "score": [],
         "reaction": []
     }
     
     for result in results:
+        data["user_type"].append(user_type)
         data["query"].append(query)
         data["retrieved_text"].append(result["text"])
-        data["score"].append(result["score"].item())
         data["reaction"].append(result["reaction"])
 
     dataset = Dataset.from_dict(data)
-    dataset.save_to_disk("al-ghazali-rag-retrieval-evaluation")
+    dataset.save_to_disk("HumbleBeeAI/al-ghazali-rag-retrieval-evaluation")
 
 def main():
     st.title("Semantic Text Retrieval Evaluation Interface")
@@ -46,6 +46,13 @@ def main():
         df, chunk_embeddings = load_data(database_file)
         st.success("Database loaded successfully!")
 
+        # Select user type
+        user_type = st.radio(
+            "Select your user type:",
+            ["Layman", "Enthusiast", "Ustaz (Expert)"],
+            horizontal=True
+        )
+
         query = st.text_area("Enter your query:")
 
         if st.button("Search") and query:
@@ -66,21 +73,21 @@ def main():
 
             results = []
 
-            for score, idx in zip(top_results_dot_product[0], top_results_dot_product[1]):
+            for _, idx in zip(top_results_dot_product[0], top_results_dot_product[1]):
                 text = df.iloc[int(idx)]["ext"]
-                st.write(f"### Score: {score:.4f}")
                 st.write(f"**Text:** {text}")
 
                 reaction = st.radio(
                     label="Rate this result:",
                     options=["👎", "🤷", "👍"],
+                    index=1,  # Default to neutral emoji
                     key=f"reaction_{idx}",
                 )
 
-                results.append({"text": text, "score": score, "reaction": reaction})
+                results.append({"text": text, "reaction": reaction})
 
             if st.button("Save Reactions"):
-                save_reactions_to_dataset(query, results)
+                save_reactions_to_dataset(user_type, query, results)
                 st.success("Reactions saved successfully!")
 
     except Exception as e: