Update app.py

structured output puzzle and math

app.py

@@ -1,4 +1,3 @@
-from ast import Interactive
 import os
 import gradio as gr
 # import openai
@@ -16,6 +15,10 @@ import base64
 import json
 from PIL import Image
 from io import BytesIO
+from pydantic import BaseModel
+import pprint
+import flatlatex
+lconv = flatlatex.converter()
 
 load_dotenv(override=True)
 key = os.getenv('OPENAI_API_KEY')
@@ -42,6 +45,30 @@ client = OpenAI(api_key = key)
 
 abbrevs = {'St. ' : 'Saint ', 'Mr. ': 'mister ', 'Mrs. ':'mussus ', 'Mr. ':'mister ', 'Ms. ':'mizz '}
 
+class Step(BaseModel):
+    explanation: str
+    output: str
+
+class MathReasoning(BaseModel):
+    steps: list[Step]
+    final_answer: str
+    
+
+def solve(prompt, chatType):
+    if chatType == 'math':
+        instruction = "You are a helpful math tutor. Guide the user through the solution step by step."
+    elif chatType == "logic":
+        instruction = "you are a helpful tutor expert in logic.  Guide the user through the solution step by step"
+    completion = client.beta.chat.completions.parse(
+        model = 'gpt-4o-2024-08-06',
+        messages = [
+            {"role": "system", "content": instruction},
+            {"role": "user", "content": prompt}
+        ],
+        response_format=MathReasoning,
+    )
+    return completion
+
 def genUsageStats(do_reset=False):
     result = []
     ttotal4o_in = 0
@@ -172,8 +199,22 @@ def updatePassword(txt):
     password = txt.lower().strip()
     return [password, "*********"]
 
-# def setModel(val):
-#     return val
+def parse_math(txt):
+    ref = 0
+    loc = txt.find(r'\\(')
+    if loc == -1:
+        return txt
+    while (True):
+        loc2 = txt[ref:].find(r'\\)')
+        if loc2 == -1:
+            break
+        loc = txt[ref:].find(r'\\(')
+        if loc > -1:
+            loc2 += 2
+            frag = lconv.convert(txt[ref:][loc:loc2])
+            txt = txt[:loc+ref] + frag + txt[loc2+ref:]
+            ref = len(txt[ref:loc]) + len(frag)
+    return txt
 
 def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_image_file=''):
     image_gen_model = 'gpt-4o-2024-08-06'
@@ -200,9 +241,20 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
             response = f'{log_cnt} log files\n{wav_cnt} .wav files\n{other_cnt} Other files:\n{others}\nlogs: {str(log_list)}'
             return [past, response, None, gptModel, uploaded_image_file]
     if user_window in unames and pwd_window == pwdList[unames.index(user_window)]:
+        chatType = 'normal'
+        prompt = prompt.strip()
+        if prompt.startswith('solve'):
+            prompt = 'How do I solve ' + prompt[5:] + ' Do not use Latex for math expressions.'
+            chatType = 'math'
+        elif prompt.startswith('puzzle'):
+            chatType = 'logic'
+            prompt = prompt[6:]
         past.append({"role":"user", "content":prompt})
         gen_image = (uploaded_image_file != '')
-        if not gen_image:
+        if chatType in ['math', 'logic']:
+           completion = solve(prompt, chatType)
+           reporting_model = image_gen_model
+        elif not gen_image:
             completion = client.chat.completions.create(model=gptModel,
                                             messages=past)
             reporting_model = gptModel
@@ -212,7 +264,17 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
             reporting_model = image_gen_model
             if not msg == 'ok':
                 return [past, msg, None, gptModel, uploaded_image_file]
-        reply = completion.choices[0].message.content
+        if chatType in ['math', 'logic']:
+            dr = completion.choices[0].message.parsed.model_dump()
+            reply = pprint.pformat(dr)
+            # df = {'final_answer' : parse_math(dr['final_answer'])}
+            # df['steps'] = []
+            # for x in dr['steps']:
+            #     df['steps'].append({'explanation': parse_math(x['explanation']), 'output' : parse_math(x['output'])})
+
+            # reply = pprint.pformat(df)
+        else:
+            reply = completion.choices[0].message.content
         tokens_in = completion.usage.prompt_tokens
         tokens_out = completion.usage.completion_tokens
         tokens = completion.usage.total_tokens
@@ -503,7 +565,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         for s in words_in:
             s = s.lstrip('- *@#$%^&_=+-')
             if len(s) > 0:
-                loc = s.index(' ')
+                loc = s.find(' ')
                 if loc > 1:
                     val = s[0:loc]
                     isnum = val.replace('.','0').isdecimal()