Update app.py

Fixed chat changed state var model after image analysis chat

app.py

@@ -176,6 +176,7 @@ def updatePassword(txt):
 #     return val
 
 def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_image_file=''):
+    image_gen_model = 'gpt-4o-2024-08-06'
     user_window = user_window.lower().strip()
     isBoss = False
     if user_window == unames[0] and pwd_window == pwdList[0]:
@@ -200,13 +201,15 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
             return [past, response, None, gptModel, uploaded_image_file]
     if user_window in unames and pwd_window == pwdList[unames.index(user_window)]:
         past.append({"role":"user", "content":prompt})
-        if uploaded_image_file == '':
+        gen_image = (uploaded_image_file != '')
+        if not gen_image:
             completion = client.chat.completions.create(model=gptModel,
                                             messages=past)
+            reporting_model = gptModel
         else:
-            gptModel = 'gpt-4o-2024-08-06'
-            (completion, msg) = analyze_image(user_window, gptModel)
+            (completion, msg) = analyze_image(user_window, image_gen_model)
             uploaded_image_file= ''
+            reporting_model = image_gen_model
             if not msg == 'ok':
                 return [past, msg, None, gptModel, uploaded_image_file]
         reply = completion.choices[0].message.content
@@ -215,7 +218,7 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
         tokens = completion.usage.total_tokens
         response += "\n\nYOU: " + prompt + "\nGPT: " + reply
         if isBoss:
-            response += f"\n{gptModel}: tokens in/out = {tokens_in}/{tokens_out}"
+            response += f"\n{reporting_model}: tokens in/out = {tokens_in}/{tokens_out}"
         if tokens > 40000:
             response += "\n\nTHIS DIALOG IS GETTING TOO LONG. PLEASE RESTART CONVERSATION SOON."
         past.append({"role":"assistant", "content": reply})
@@ -225,7 +228,7 @@ def chat(prompt, user_window, pwd_window, past, response, gptModel, uploaded_ima
                 dataFile = new_func(user_window)
                 with open(dataFile, 'a') as f:
                     m = '4o'
-                    if 'mini' in gptModel:
+                    if 'mini' in reporting_model:
                         m = '4omini'
                     f.write(f'{user_window}:{tokens_in}/{tokens_out}-{m}\n')
                 accessOk = True