Upload folder using huggingface_hub

server/mentalHealthPatientenv_environment.py

@@ -121,7 +121,7 @@ class MentalhealthpatientenvEnvironment(Environment):
         elif action.action_type == "diagnose":
             # Simulate a diagnosis action by providing feedback based on the patient's state
                 self._state.action_sequence.append(action.action_type)
-                curr_diagnosis = action.message.strip().lower()
+                curr_diagnosis=action.message.strip().split(',')
                 actual_disorders = self._state.disorder
                 
                 reward = calculate_reward(

server/reward.py

@@ -119,14 +119,28 @@ def calculate_reward(
         # Diagnosis score
         diagnosis_score = 0.0
         if action_type == "diagnose" and diagnosis:
-            similarity = disorder_similarity(disorder, diagnosis)
+            if len(diagnosis)==1:
+                similarity = disorder_similarity(disorder, diagnosis[-1])
 
-            if similarity == "full":
-                diagnosis_score = 1.0
-            elif similarity == "partial":
-                diagnosis_score = 0.5
+                if similarity == "full":
+                    diagnosis_score = 1.0
+                elif similarity == "partial":
+                    diagnosis_score = 0.5
+                else:
+                    diagnosis_score = 0.0
             else:
-                diagnosis_score = 0.0
+                diagnosis_score=0.0
+                for d in diagnosis:
+                    similarity=disorder_similarity(disorder,d)
+
+                    if similarity=="full":
+                        diagnosis_score+=0.5
+                    elif diagnosis_score=="partial":
+                        diagnosis_score+=0.3
+                    else:
+                        diagnosis_score-=0.5
+                diagnosis_score=normalize(diagnosis_score,0.0,1.0)
+
 
         # Repetition penalty (converted to positive scale)
         repeat_penalty = 0.0