Upload folder using huggingface_hub

.idea/workspace.xml

@@ -120,7 +120,7 @@
       <workItem from="1710618477574" duration="1782000" />
       <workItem from="1710851879119" duration="642000" />
       <workItem from="1710940809173" duration="1193000" />
-      <workItem from="1711350259553" duration="2760000" />
+      <workItem from="1711350259553" duration="3999000" />
     </task>
     <task id="LOCAL-00001" summary="update README.md">
       <option name="closed" value="true" />
@@ -153,6 +153,6 @@
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
     <SUITE FILE_PATH="coverage/llama2_webui$app.coverage" NAME="app Coverage Results" MODIFIED="1708306249728" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/wellbee_webui$app.coverage" NAME="app Coverage Results" MODIFIED="1711352680217" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="" />
+    <SUITE FILE_PATH="coverage/wellbee_webui$app.coverage" NAME="app Coverage Results" MODIFIED="1711355116443" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="" />
   </component>
 </project>

app.py

@@ -49,6 +49,8 @@ data = [
     {"emotion": "neutral", "value": 0},
 ]
 
+default_therapist = False
+
 
 def get_emotion_value(value):
     global num_questions
@@ -68,7 +70,6 @@ def sort_data(d):
 
 emotion_data = sort_data(data)
 
-
 DEFAULT_SYSTEM_PROMPT = """You are a helpful and truthful psychology and psychotherapy assistant. With each 
     response, you offer follow-up questions to encourage openness and continue the conversation naturally. Use 
     compassionate listening to have helpful and meaningful conversations with users. You are empathic and friendly. 
@@ -84,6 +85,7 @@ DEFAULT_SYSTEM_PROMPT = """You are a helpful and truthful psychology and psychot
     supportive and understanding environment for users to share their feelings and concerns. Always prioritize their 
     well-being and safety."""
 
+
 def main():
     classifier = pipeline(task="text-classification", model="SamLowe/roberta-base-go_emotions", top_k=None)
     parser = argparse.ArgumentParser()
@@ -111,8 +113,6 @@ def main():
 
     load_dotenv()
 
-
-
     DEFAULT_SYSTEM_PROMPT_THERAPIST = """You are a helpful and truthful psychology and psychotherapy assistant. With each 
         response, you offer follow-up questions to encourage openness and continue the conversation naturally. Use 
         compassionate listening to have helpful and meaningful conversations with users. Keep your answers short (max 3 
@@ -149,7 +149,7 @@ def main():
     supportive and understanding environment for users to share their feelings and concerns. Always prioritize their 
     well-being and safety."""
 
-    DEFAULT_SYSTEM_PROMPT_FRIEND="""
+    DEFAULT_SYSTEM_PROMPT_FRIEND = """
     For the rest of this chat, you must pretend to be my closest friend. Your goal is to sound like someone similar aged
      to the user and respond in short messages that feel like texting. Start the conversation with a short casual 
      greeting, and keep your messages brief and upbeat. Use some abbreviations to add personality to your messages and 
@@ -262,7 +262,6 @@ def main():
             message = ""
         return history, message or ""
 
-
     def generate(
             message: str,
             history_with_input: list[tuple[str, str]],
@@ -276,7 +275,7 @@ def main():
 
         emotion_insights_flag = False
 
-        if message=="Emotion Insights XXXXX":
+        if message == "Emotion Insights XXXXX":
             emotion_insights_flag = True
             system_prompt = """
             You are a supportive and understanding life and personal growth coach. You analyse the messages and 
@@ -310,7 +309,8 @@ def main():
             # print(message)
         else:
             system_prompt = DEFAULT_SYSTEM_PROMPT
-        # print("System prompt:", system_prompt)
+
+        print("System prompt:", system_prompt)
 
         if max_new_tokens > MAX_MAX_NEW_TOKENS:
             raise ValueError
@@ -359,17 +359,18 @@ def main():
     default_advanced_checkbox = False
 
     default_show_emotions = False
-    default_therapist = False
+
     def convert_summary_to_prompt(summary):
         return prompts_container.get_prompt_by_summary(summary)
 
     def toggle_action(value):
-        global DEFAULT_SYSTEM_PROMPT
-        if value:
+        global DEFAULT_SYSTEM_PROMPT, default_therapist
+        default_therapist = not default_therapist
+        if default_therapist:
             DEFAULT_SYSTEM_PROMPT = DEFAULT_SYSTEM_PROMPT_FRIEND
         else:
             DEFAULT_SYSTEM_PROMPT = DEFAULT_SYSTEM_PROMPT_THERAPIST
-        debug = "Toggle is **on**." if value else "Toggle is **off**."
+        debug = "Toggle is Friend." if default_therapist else "Toggle is Therapist."
         print(debug)
 
     def two_columns_list(tab_data, chatbot):
@@ -446,7 +447,6 @@ def main():
                     )
                     # toggle = Toggle(label="Therapist/Friend", show_label=True, info="Change between Therapist and Friend")
                 with gr.Group():
-
                     initial_message = """
                     Hello, there! I'm your virtual mental health assistant, here to provide support and guidance on your journey towards better mental well-bee. 
                     It takes a lot of courage to open up and talk about what's on your mind, and I want to thank you for trusting me with that.
@@ -622,11 +622,11 @@ def main():
         )
 
         report_button.click(
-                fn=clear_and_save_textbox1,
-                outputs=[textbox, saved_input],
-                api_name=False,
-                queue=False,
-            ).then(
+            fn=clear_and_save_textbox1,
+            outputs=[textbox, saved_input],
+            api_name=False,
+            queue=False,
+        ).then(
             fn=generate,
             inputs=[
                 saved_input,
@@ -733,9 +733,9 @@ def main():
             api_name=False,
         )
 
-
     # share=args.share
     demo.queue(max_size=20).launch(share=False, show_api=False)
 
+
 if __name__ == "__main__":
     main()