better logging

app.py

@@ -2,6 +2,7 @@ import base64
 import streamlit as st
 from openai import OpenAI
 import os
+from utils import pprint
 
 # Load environment variables
 from dotenv import load_dotenv
@@ -56,6 +57,9 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Initialize session state
+if "ipAddress" not in st.session_state:
+    st.session_state.ipAddress = st.context.headers.get("x-forwarded-for")
+
 if 'cooking_equipment' not in st.session_state:
     st.session_state.cooking_equipment = {
         'Stove': True,
@@ -120,12 +124,27 @@ def analyze_and_generate_recipe(uploaded_image, available_equipment=None, langua
                     ]
                 }
             ]
+
+            # Log API call details
+            pprint({
+                "function": "analyze_and_generate_recipe",
+                "model": "gpt-4o-mini",
+                "language": language,
+                "available_equipment": available_equipment
+            })
             
             # Make the LLM call
             response = client.chat.completions.create(
                 model="gpt-4o-mini",
                 messages=messages
             )
+
+            # Log response details
+            pprint({
+                "function": "analyze_and_generate_recipe_response",
+                "tokens_used": response.usage.total_tokens if response.usage else None,
+                "response_length": len(response.choices[0].message.content)
+            })
             
             return response.choices[0].message.content
         except Exception as e:
@@ -136,6 +155,14 @@ def analyze_and_generate_recipe(uploaded_image, available_equipment=None, langua
 def refine_recipe(original_recipe, user_refinement, language='English'):
     """Refine the recipe based on user input"""
     try:
+        # Log API call details
+        pprint({
+            "function": "refine_recipe",
+            "model": "gpt-4o-mini",
+            "language": language,
+            "user_refinement_length": len(user_refinement)
+        })
+
         response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
@@ -156,6 +183,13 @@ def refine_recipe(original_recipe, user_refinement, language='English'):
                 }
             ]
         )
+
+        # Log response details
+        pprint({
+            "function": "refine_recipe_response",
+            "tokens_used": response.usage.total_tokens if response.usage else None,
+            "response_length": len(response.choices[0].message.content)
+        })
         
         return response.choices[0].message.content
     except Exception as e:

utils.py

@@ -0,0 +1,13 @@
+import datetime as DT
+import pytz
+import streamlit as st
+
+
+def __nowInIST() -> DT.datetime:
+    return DT.datetime.now(pytz.timezone("Asia/Kolkata"))
+
+
+def pprint(log: str):
+    now = __nowInIST()
+    now = now.strftime("%Y-%m-%d %H:%M:%S")
+    print(f"[{now}] [{st.session_state.ipAddress}] {log}")