Update app.py

app.py

@@ -48,9 +48,17 @@ def TopicClassification(payload: TextIn):
     result = nlp(text, truncation=True)
     return {"result": result}
 
-
+# https://hinacortus-api-knowyouraudience.hf.space/whichsocial/myspace
 @app.get("/whichsocial/{request}")
 def whichsocial(request):
+    if ('http' in request or 'https' in request or 'www' in request or '.com' in request or '.fr' in request:
+        request = "website"
+        listsocialnetwork = ['facebook', 'youtube', 'myspace', 'linkedin', 'twitter', 'instagram', 'github',
+                            'reddit', 'picterest', 'discord', '']
+        link_split = request.split('/')
+        for part in link_split:
+            if part in listsocialnetwork:
+                website = "found"
     socialnetwork = 'TikTok'
     userprofile = 'me'
-    return {"request": request, "social_network": socialnetwork, "user_profile": userprofile}
+    return {"request": request, "website":website, "social_network": socialnetwork, "user_profile": userprofile}