Update run.py

run.py

@@ -80,12 +80,20 @@ collection = client.get_collection(name="chromaTS", embedding_function=sentence_
 
 inference_client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def extract_ip_and_device(headers):
+def extract_ip_and_device(headers_obj):
     ip_address = None
     device_info = None
-
+    
+    # Access the raw headers list
+    headers = headers_obj.raw
+    
     for header in headers:
+        if len(header) != 2:
+            print(f"Unexpected header format: {header}")
+            continue
+        
         key, value = header
+        
         if key == b'x-forwarded-for':
             ip_address = value.decode('utf-8')
         elif key == b'user-agent':