Update app.py

Added latex math handling to deepseek models

app.py

@@ -24,6 +24,7 @@ import pytz
 import math
 import numpy as np
 # import matplotlib.pyplot as plt
+from pylatexenc.latex2text import LatexNodes2Text
 
 
 load_dotenv(override=True)
@@ -90,6 +91,31 @@ def date_from_utime(utime):
     eastern = pytz.timezone('US/Eastern')
     return dt.astimezone(eastern).strftime('%Y-%m-%d')
 
+def convert_latex_math(text):
+    lines = text.split('\n')
+    start_line = False
+    out_txt = ''
+    for line in lines:
+        if len(line) == 0:
+            out_txt += '\n'
+            continue
+        else:
+            if line == r'\]':
+                continue
+            if line == r'\[':
+                start_line = True
+                continue
+            if start_line:
+                line = '\n' + LatexNodes2Text().latex_to_text(line.strip())
+                start_line = False
+            if line.startswith(r'\['):
+                loc = line.find(r'\]')
+                if loc > 0:
+                    latex_code = line[2:loc]
+                    line = '\n' + LatexNodes2Text().latex_to_text(latex_code)
+        out_txt += (line + '\n')
+    return out_txt
+
 def stock_list():
     rv = ''
     with open(stock_data_path, 'rt') as fp:
@@ -698,6 +724,8 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
         elif prompt.lower().startswith('puzzle'):
             chatType = 'logic'
             prompt = prompt[6:]
+        if deepseek:
+            prompt = prompt + '. Do not use Latex for math expressions.'
         past.append({"role":"user", "content":prompt})
         gen_image = (uploaded_image_file != '')
         if chatType in special_chat_types:
@@ -732,6 +760,9 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
                 return [past, msg, None, gptModel, uploaded_image_file, plot]
         if not chatType in special_chat_types:
             reply = completion.choices[0].message.content
+            # if 'groq' in reporting_model:
+            if deepseek:
+                reply = convert_latex_math(reply)
             final_text = reply
             if deepseek:
                 loc1 = reply.find('<think>')
@@ -739,11 +770,11 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
                     loc2 = reply.find('</think>')
                     if loc2 > loc1:
                         final_text = reply[loc2 + 8:]
-                reply = reply.replace('<think>','Thinking:\n').replace('</think>','Done thinking:\n')
+                reply = reply.replace('<think>','\n***Thinking***\n').replace('</think>','\n***Done thinking***\n')
             tokens_in = completion.usage.prompt_tokens
             tokens_out = completion.usage.completion_tokens
             tokens = completion.usage.total_tokens
-        response += md("\n\nYOU: " + prompt + "\nGPT: " + reply)
+        response += md("\n\n***YOU***: " + prompt + "\n***GPT***: " + reply)
         if isBoss:
             response += md(f"\n{reporting_model}: tokens in/out = {tokens_in}/{tokens_out}")
         if tokens > 40000: