emotional stage modification in re

app.py

@@ -52,44 +52,46 @@ def emotion_analysis(user_input):
 
     # Step 2: Generate LLM response
     prompt = f"""
-    Task: You are a social psychologist specializing in Roy Baumeister's six-stage theory of emotional progression. Your task is to analyze emotional states based on user input while adhering to specific response boundaries.
+    Task: You are a social psychologist specializing in Roy Baumeister's six-stage theory of emotional progression. Your task is to analyze emotional states based on user input while adhering strictly to specific response boundaries.
 
     [Input Information]:
     **User Input**: "{user_input}"
     **Model Output**: "{predicted_emotion}"
 
     Specifics:
-    1. Strictly respond only to questions or input related to mental health or emotional well-being. For unrelated input, reply with: "Not a valid question." 
-       - Example: If the user asks about weather, sports, or other unrelated topics, respond with: "Not a valid question."
-    2. Use the **User Input** as the primary source for determining the emotional state, while considering the **Model Output** ("happy" or "depressed") only as a secondary reference.
-    3. Assign the user’s emotional state to one of Roy Baumeister’s six stages of emotional progression:
+    1. Respond **only** to questions or input related to mental health or emotional well-being. For unrelated input, reply strictly with: "Not a valid question."
+    2. Use the **User Input** as the primary source for determining the emotional state, while considering the **Model Output** ("happy" or "depressed") as a secondary reference.
+    3. Categorize the user’s emotional state into one of Roy Baumeister’s six stages of emotional progression:
        - Stage 1: Falling short of expectations
        - Stage 2: Attributions to self
        - Stage 3: High self-awareness
        - Stage 4: Negative affect
        - Stage 5: Cognitive deconstruction
        - Stage 6: Disinhibition
-    4. Provide specific recommendations for the assigned stage:
-       - If the user is **depressed**, suggest stage-specific remedies to improve their emotional state.
-       - If the user is **happy**, suggest strategies to maintain or enhance their happiness.
-    5. Prioritize clarity, empathy, and practicality in your analysis and suggestions.
+    4. Provide specific stage-based recommendations:
+       - If **depressed**, suggest remedies for improving emotional state.
+       - If **happy**, suggest strategies for maintaining or enhancing happiness.
+    5. Maintain clarity, empathy, and practicality in all analyses and suggestions.
 
     [Response Rules]:
-    - Do NOT attempt to provide an output if the input is not related to mental health.
-    - Always analyze the user’s input independently, even if it conflicts with the model’s predicted output.
+    - If input is unrelated to mental health, reply only with: "Not a valid question."
+    - Always analyze user input independently, even if it conflicts with the model's predicted output.
 
-    [Desired Output Format]:
+    [Desired Output Format for valid input]:
     Emotional Analysis:
     I'd say you're feeling: <Happy/Depressed>
     Emotional Stage: <Stage and brief reasoning>
     Suggested Remedies/Strategies: <Practical advice based on the stage>
+
+    For invalid input:
+    "Not a valid question."
     """
 
     try:
         completion = client.chat.completions.create(
             model="nvidia/nemotron-4-340b-instruct",
             messages=[{"role": "user", "content": prompt}],
-            temperature=0.2,
+            temperature=0.5,
             top_p=0.7,
             max_tokens=512,
             stream=True
@@ -114,6 +116,7 @@ def emotion_analysis(user_input):
 def extract_analysis_details(analysis_text):
     feelings_match = re.search(r"I'd say you're feeling:\s*([^\n]+)", analysis_text)
     feelings = feelings_match.group(1).strip() if feelings_match else "Not Found"
+    print(feelings)
     if feelings.lower() == "happy":
         feelings = feelings + " with Positive Mindset"
     elif feelings.lower() == "depressed":
@@ -122,7 +125,9 @@ def extract_analysis_details(analysis_text):
         feelings
 
     # Extract emotional stage
-    stage_match = re.search(r"Emotional Stage:\s*([^\n.]+)", analysis_text)
+    # stage_match = re.search(r"Emotional Stage:\s*([^\n.]+)", analysis_text)
+    # print(stage_match)
+    stage_match = re.search(r"Emotional Stage:\s*(.*?)(?=\n[A-Z])", analysis_text, re.DOTALL)
     emotional_stage = stage_match.group(1).strip() if stage_match else "Not Found"
 
     # Regex to match the section header and capture from there to the end
@@ -131,10 +136,11 @@ def extract_analysis_details(analysis_text):
     suggestions = match.group(0).strip() if match else "No matching section found."
     # print(suggestions)
 
-    if feelings == "Not Found":
+    if feelings == "Not Found" or feelings == "Not a valid question.":
         feelings = "Not a valid question."
     return feelings, emotional_stage, suggestions
 
+
 # Gradio interface with input validation
 def validate_and_run(user_input):
     if not user_input.strip():  # Check if the input is empty or just spaces