Update app.py

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import os
 import uuid
 import json
+import re
 
 # Page configuration
 st.set_page_config(
@@ -85,6 +86,34 @@ def get_inference_client():
         api_key=api_token,
     )
 
+# Clean the output to remove thinking process
+def clean_output(text):
+    # Extract only the parts after the headings
+    cleaned_text = ""
+    
+    # Extract Emotional tone section
+    if "Emotional tone:" in text:
+        match = re.search(r'Emotional tone:(.*?)(?:Recurring themes:|$)', text, re.DOTALL)
+        if match:
+            emotional_tone = match.group(1).strip()
+            cleaned_text += f"😊 **Emotional tone:** {emotional_tone}\n\n"
+    
+    # Extract Recurring themes section
+    if "Recurring themes:" in text:
+        match = re.search(r'Recurring themes:(.*?)(?:Advice:|$)', text, re.DOTALL)
+        if match:
+            themes = match.group(1).strip()
+            cleaned_text += f"🔍 **Recurring themes:** {themes}\n\n"
+    
+    # Extract Advice section
+    if "Advice:" in text:
+        match = re.search(r'Advice:(.*?)$', text, re.DOTALL)
+        if match:
+            advice = match.group(1).strip()
+            cleaned_text += f"💡 **Advice:** {advice}"
+    
+    return cleaned_text if cleaned_text else text
+
 # Function to analyze journal entries
 def analyze_journal(entry, model_name="deepseek-ai/DeepSeek-R1"):
     client = get_inference_client()
@@ -96,12 +125,13 @@ def analyze_journal(entry, model_name="deepseek-ai/DeepSeek-R1"):
 
 Journal entry: "{entry}"
 
-Please provide:
-1. Emotional tone: Identify the primary emotions expressed in the entry (like anxious, happy, stressed, etc.)
+Please provide the following analysis directly, without explaining your thinking process:
+
+1. Emotional tone: Identify the primary emotions expressed in the entry
 2. Recurring themes: Note any patterns or important topics mentioned
 3. Advice: Offer kind, supportive guidance tailored to what was shared
 
-Format your response clearly with these exact headings (Emotional tone, Recurring themes, Advice) but without using JSON.
+Format your response with only these three sections, using the exact headings: "Emotional tone:", "Recurring themes:", and "Advice:". Do not include any additional explanations or thinking.
 """
         
         completion = client.chat.completions.create(
@@ -118,22 +148,8 @@ Format your response clearly with these exact headings (Emotional tone, Recurrin
         
         output = completion.choices[0].message.content
         
-        # Format the output nicely
-        formatted_output = ""
-        
-        if "Emotional tone:" in output:
-            emotional_tone = output.split("Emotional tone:")[1].split("Recurring themes:")[0].strip()
-            formatted_output += f"😊 **Emotional tone:** {emotional_tone}\n\n"
-        
-        if "Recurring themes:" in output:
-            themes = output.split("Recurring themes:")[1].split("Advice:")[0].strip()
-            formatted_output += f"🔍 **Recurring themes:** {themes}\n\n"
-        
-        if "Advice:" in output:
-            advice = output.split("Advice:")[1].strip()
-            formatted_output += f"💡 **Advice:** {advice}"
-        
-        return formatted_output if formatted_output else output
+        # Clean the output to remove any thinking process
+        return clean_output(output)
         
     except Exception as e:
         st.error(f"API Error: {str(e)}")
@@ -344,7 +360,4 @@ st.markdown("---")
 st.caption("""
 **Disclaimer**: This app is for educational purposes only and is not a substitute for professional mental health advice, diagnosis, or treatment. 
 Always seek the advice of your physician or other qualified health provider with any questions regarding a medical condition.
-""")
-     
-        
-
+""")