Merge pull request #44 from dev9086/main

Increased Latency Caused by Redundant Network Requests

Author: eccentriccoder01

core/utils.py

@@ -85,6 +85,34 @@ def get_ai_response(user_message, model):
     except Exception as e:
         return "I'm here to listen and support you. Sometimes I have trouble connecting, but I want you to know that your feelings are valid and you're not alone. Would you like to share more about what you're experiencing?"
 
+def cached_user_ip():
+    # Check if IP is already cached in session state
+    if hasattr(st.session_state, 'cached_ip') and hasattr(st.session_state, 'ip_cache_time'):
+        # Check if cache is still valid (cache for 1 hour)
+        cache_age = datetime.now() - st.session_state.ip_cache_time
+        if cache_age < timedelta(hours=1):
+            return st.session_state.cached_ip
+    
+    # Cache is missing or expired, fetch new IP
+    try:
+        response = requests.get("https://api.ipify.org", timeout=5)
+        ip = response.text.strip()
+        # Cache the IP and timestamp
+        st.session_state.cached_ip = ip
+        st.session_state.ip_cache_time = datetime.now()
+        
+        return ip
+    except (requests.RequestException, requests.Timeout, Exception):
+        # Fallback: use session ID or generate a unique identifier
+        fallback_id = f"session_{hash(str(st.session_state)) % 100000}"
+        
+        # Cache the fallback ID so we use the same one consistently
+        if not hasattr(st.session_state, 'cached_ip'):
+            st.session_state.cached_ip = fallback_id
+            st.session_state.ip_cache_time = datetime.now()
+        
+        return st.session_state.cached_ip
+
 #Implementing IP Based Isolation
 def get_user_ip():
     try:
@@ -94,7 +122,7 @@ def get_user_ip():
 
 #Saving and loading to/from JSON File
 def get_memory_file():
-    ip = get_user_ip()
+    ip = cached_user_ip()
     os.makedirs("data", exist_ok=True)
     return f"data/conversations_{ip}.json"
 
@@ -108,4 +136,4 @@ def load_conversations():
     if not os.path.exists(memory_file):
         return []
     with open(memory_file, 'r', encoding="utf-8") as f:
-        return json.load(f)
+        return json.load(f)