Update app.py

app.py

@@ -8,13 +8,17 @@ from jwt import ExpiredSignatureError, InvalidTokenError
 from starlette import status
 from functions import *
 import pandas as pd
-from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi import FastAPI, File, UploadFile, HTTPException,Request
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from src.api.speech_api import speech_translator_router
 from functions import client as supabase
 from urllib.parse import urlparse
 import nltk
+from collections import Counter, defaultdict
+from datetime import datetime, timedelta
+from dateutil.parser import isoparse
+
 
 
 nltk.download('punkt_tab')
@@ -526,4 +530,139 @@ async def chatHistory(vectorstore: str):
 # @app.post("/trainChatbot")
 # async def chatHistory(vectorstore: str):
 #     username, chatbotName = vectorstore.split("$")[1], vectorstore.split("$")[2]
-#     return response
+#     return response
+
+def get_ip_info(ip: str):
+    try:
+        response = requests.get(f"https://ipinfo.io/{ip}/json")
+        data = response.json()
+        return data.get("city", "Unknown")
+    except Exception as e:
+        return "Unknown"
+
+
+@app.post("/daily_chat_count")
+async def daily_chat_count(days: int = 7):
+    response = supabase.table("ConversAI_ChatHistory").select("*").execute().data
+
+    seven_days_ago = datetime.now().astimezone() - timedelta(days=days)
+    dates = [
+        isoparse(i["timestamp"]).date()
+        for i in response
+        if isoparse(i["timestamp"]) >= seven_days_ago
+    ]
+
+    date_count = Counter(dates)
+
+    data = [{"date": date.isoformat(), "count": count} for date, count in date_count.items()]
+
+    return {"data": data}
+
+
+@app.post("/daily_active_end_user")
+async def daily_active_end_user(days: int = 7):
+    response = supabase.table("ConversAI_ChatHistory").select("*").execute().data
+
+    seven_days_ago = datetime.now().astimezone() - timedelta(days=days)
+
+    ip_by_date = defaultdict(set)
+
+    for i in response:
+        timestamp = isoparse(i["timestamp"])
+        ip_address = i["IpAddress"]
+        if timestamp >= seven_days_ago:
+            date = timestamp.date()
+            ip_by_date[date].add(ip_address)
+
+    data = [{"date": date.isoformat(), "terminal": len(ips)} for date, ips in ip_by_date.items() if len(ips) > 1]
+
+    return {"data": data}
+
+
+@app.post("/average_session_interaction")
+async def average_session_interaction(days: int = 7):
+    response = supabase.table("ConversAI_ChatHistory").select("*").execute().data
+
+    seven_days_ago = datetime.now().astimezone() - timedelta(days=days)
+
+    total_messages_by_date = defaultdict(int)
+    unique_ips_by_date = defaultdict(set)
+
+    for i in response:
+        timestamp = isoparse(i["timestamp"])
+        ip_address = i["IpAddress"]
+        if timestamp >= seven_days_ago:
+            date = timestamp.date()
+            total_messages_by_date[date] += 1
+            unique_ips_by_date[date].add(ip_address)
+
+    data = []
+    for date in sorted(total_messages_by_date.keys()):
+        total_messages = total_messages_by_date[date]
+        unique_ips = len(unique_ips_by_date[date])
+        average_interactions = total_messages / unique_ips if unique_ips > 0 else 0
+        data.append({"date": date.isoformat(), "interactions": average_interactions})
+
+    return {"data": data}
+
+
+@app.post("/token_usages")
+async def token_usages(days: int = 7):
+    response = supabase.table("ConversAI_ChatHistory").select("*").execute().data
+
+    seven_days_ago = datetime.now().astimezone() - timedelta(days=days)
+
+    token_usage_by_date = defaultdict(int)
+
+    for i in response:
+        timestamp = isoparse(i["timestamp"])
+        if timestamp >= seven_days_ago:
+            date = timestamp.date()
+            response_token_count = i.get("ResponseTokenCount")
+            if response_token_count is not None:
+                token_usage_by_date[date] += response_token_count
+
+    data = [{"date": date.isoformat(), "total_tokens": total_tokens} for date, total_tokens in
+            token_usage_by_date.items()]
+
+    return {"data": data}
+
+
+@app.post("/add_feedback")
+async def add_feedback(request: Request, feedback: str, user_id: str):
+    client_ip = request.client.host
+    city = get_ip_info(client_ip)
+
+    response = supabase.table("ConversAI_Feedback").insert(
+        {"feedback": feedback, "user_id": user_id, "city": city, "ip": client_ip}).execute()
+
+    return {"message": "success"}
+
+
+@app.post("/user_satisfaction_rate")
+async def user_satisfaction_rate():
+    response = supabase.table("ConversAI_Feedback").select("*").execute().data
+
+    seven_days_ago = datetime.now().astimezone() - timedelta(days=7)
+
+    feedback_counts = defaultdict(lambda: {"like": 0, "dislike": 0})
+
+    for i in response:
+        timestamp = isoparse(i["timestamp"])
+        if timestamp >= seven_days_ago:
+            date = timestamp.date()
+            feedback = i.get("feedback")
+            if feedback == "like":
+                feedback_counts[date]["like"] += 1
+            elif feedback == "dislike":
+                feedback_counts[date]["dislike"] += 1
+
+    data = []
+    for date in sorted(feedback_counts.keys()):
+        like_count = feedback_counts[date]["like"]
+        dislike_count = feedback_counts[date]["dislike"]
+        total_feedback = like_count + dislike_count
+        satisfaction_rate = (like_count / total_feedback * 100) if total_feedback > 0 else 0
+        data.append({"date": date.isoformat(), "rate": satisfaction_rate})
+
+    return {"data": data}