Update run.py

run.py

@@ -80,8 +80,18 @@ collection = client.get_collection(name="chromaTS", embedding_function=sentence_
 
 inference_client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# Global variable to store the URL
-global_url = ""
+def extract_ip_and_device(headers):
+    ip_address = None
+    device_info = None
+
+    for header in headers:
+        key, value = header
+        if key == b'x-forwarded-for':
+            ip_address = value.decode('utf-8')
+        elif key == b'user-agent':
+            device_info = value.decode('utf-8')
+    
+    return ip_address, device_info
 
 def format_prompt(message, history):
     print("HISTORY")
@@ -112,6 +122,7 @@ def response(request: gr.Request,prompt, history, temperature=0.9, max_new_token
 
     print(f"Working with URL: {url_extracted}")
     headers = request.headers
+    IP, dev = extract_ip_and_device(headers)
     print(headers)
     temperature = float(temperature)
     if temperature < 1e-2: temperature = 1e-2
@@ -143,7 +154,7 @@ def response(request: gr.Request,prompt, history, temperature=0.9, max_new_token
         output += response.token.text
         yield output
     now = str(datetime.now())
-    save_to_sheet(now, prompt, output)
+    save_to_sheet(now, prompt, output, IP, dev, headers)
     yield output
     
 gr.ChatInterface(