Spaces:
Running
Running
File size: 2,468 Bytes
14aa01b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
import os
import requests
import json
import re
import streamlit as st
from langdetect import detect_langs
from langcodes import Language
headers = {
'Content-Type': 'application/json',
'Accept': '*/*'
}
def identify_language(response):
lang_code = detect_langs(response)[0].lang
return Language.make(language=lang_code).display_name()
def thumbs_feedback(feedback, **kwargs):
"""
Sends feedback to Amplitude Analytics
"""
send_amplitude_data(
user_query=kwargs.get("user_query", "No user input"),
bot_response=kwargs.get("bot_response", "No bot response"),
demo_name=kwargs.get("demo_name", "Unknown"),
feedback=feedback['score'],
)
st.session_state.feedback_key += 1
def send_amplitude_data(user_query, bot_response, demo_name, feedback=None):
# Send query and response to Amplitude Analytics
data = {
"api_key": os.getenv('AMPLITUDE_TOKEN'),
"events": [{
"device_id": st.session_state.device_id,
"event_type": "submitted_query",
"event_properties": {
"Space Name": demo_name,
"Demo Type": "Agent",
"query": user_query,
"response": bot_response,
"Response Language": identify_language(bot_response)
}
}]
}
if feedback:
data["events"][0]["event_properties"]["feedback"] = feedback
response = requests.post('https://api2.amplitude.com/2/httpapi', headers=headers, data=json.dumps(data))
if response.status_code != 200:
print(f"Amplitude request failed with status code {response.status_code}. Response Text: {response.text}")
def escape_dollars_outside_latex(text):
# Define a regex pattern to find LaTeX equations (either single $ or double $$)
pattern = re.compile(r'(\$\$.*?\$\$|\$.*?\$)')
latex_matches = pattern.findall(text)
# Placeholder to temporarily store LaTeX equations
placeholders = {}
for i, match in enumerate(latex_matches):
placeholder = f'__LATEX_PLACEHOLDER_{i}__'
placeholders[placeholder] = match
text = text.replace(match, placeholder)
# Escape dollar signs in the rest of the text
text = text.replace('$', '\\$')
# Replace placeholders with the original LaTeX equations
for placeholder, original in placeholders.items():
text = text.replace(placeholder, original)
return text
|