Upload 6 files

Dockerfile

@@ -0,0 +1,21 @@
+FROM python:3.9
+
+WORKDIR /code 
+
+COPY ./requirements.txt /code/requirements.txt 
+
+RUN pip install --upgrade pip
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY ./__init__.py /code/__init__.py
+
+COPY ./.env /code/.env 
+
+COPY ./mini_data.csv /code/mini_data.csv 
+
+COPY ./recommend.py /code/recommend.py 
+
+COPY ./main.py /code/main.py  
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,28 @@
+from fastapi import FastAPI
+from recommend import vector_search
+from pydantic import BaseModel
+import requests 
+import pandas as pd
+
+df = pd.read_csv('mini_data.csv')
+
+
+class validation(BaseModel):
+    title: str 
+
+
+app = FastAPI()
+
+
+@app.post('/recommend')
+def get_recommendations(items: validation):
+    anime = df[df['title'] == items.title]
+
+    if len(anime) == 0:
+        return None 
+    
+    desc = anime['description']
+
+    completion = vector_search(desc)
+
+    return completion

mini_data.csv

@@ -0,0 +1,55 @@
+,id,url,trailer,title,number_of_episodes,status,is_airing,duration_per_episode,rating,score,rank,popularity,description,,genres,themes,streaming_sites
+0,1,https://myanimelist.net/anime/1/Cowboy_Bebop,https://www.youtube.com/watch?v=gY5nDXOtv_o,Cowboy Bebop,26,Finished Airing,False,24 min per ep,R - 17+ (violence & profanity),8.75,44,43,"Crime is timeless. By the year 2071, humanity has expanded across the galaxy, filling the surface of other planets with settlements like those on Earth. These new societies are plagued by murder, drug use, and theft, and intergalactic outlaws are hunted by a growing number of tough bounty hunters.
+
+Spike Spiegel and Jet Black pursue criminals throughout space to make a humble living. Beneath his goofy and aloof demeanor, Spike is haunted by the weight of his violent past. Meanwhile, Jet manages his own troubled memories while taking care of Spike and the Bebop, their ship. The duo is joined by the beautiful con artist Faye Valentine, odd child Edward Wong Hau Pepelu Tivrusky IV, and Ein, a bioengineered Welsh Corgi.
+
+While developing bonds and working to catch a colorful cast of criminals, the Bebop crew's lives are disrupted by a menace from Spike's past. As a rival's maniacal plot continues to unravel, Spike must choose between life with his newfound family or revenge for his old wounds.
+
+[Written by MAL Rewrite]",1998.0,"[{'mal_id': 1, 'type': 'anime', 'name': 'Action', 'url': 'https://myanimelist.net/anime/genre/1/Action'}, {'mal_id': 46, 'type': 'anime', 'name': 'Award Winning', 'url': 'https://myanimelist.net/anime/genre/46/Award_Winning'}, {'mal_id': 24, 'type': 'anime', 'name': 'Sci-Fi', 'url': 'https://myanimelist.net/anime/genre/24/Sci-Fi'}]","[{'mal_id': 50, 'type': 'anime', 'name': 'Adult Cast', 'url': 'https://myanimelist.net/anime/genre/50/Adult_Cast'}, {'mal_id': 29, 'type': 'anime', 'name': 'Space', 'url': 'https://myanimelist.net/anime/genre/29/Space'}]","[{'name': 'Crunchyroll', 'url': 'http://www.crunchyroll.com/series-271225'}, {'name': 'Funimation', 'url': 'https://www.funimation.com/shows/cowboy-bebop'}, {'name': 'Netflix', 'url': 'https://www.netflix.com/title/80001305'}, {'name': 'Tubi TV', 'url': 'https://tubitv.com/series/2052/cowboy-bebop-subtitled'}]"
+1,5,https://myanimelist.net/anime/5/Cowboy_Bebop__Tengoku_no_Tobira,,Cowboy Bebop: Tengoku no Tobira,1,Finished Airing,False,1 hr 55 min,R - 17+ (violence & profanity),8.38,193,612,"Another day, another bounty—such is the life of the often unlucky crew of the Bebop. However, this routine is interrupted when Faye, who is chasing a fairly worthless target on Mars, witnesses an oil tanker suddenly explode, causing mass hysteria. As casualties mount due to a strange disease spreading through the smoke from the blast, a whopping three hundred million woolong price is placed on the head of the supposed perpetrator.
+
+With lives at stake and a solution to their money problems in sight, the Bebop crew springs into action. Spike, Jet, Faye, and Edward, followed closely by Ein, split up to pursue different leads across Alba City. Through their individual investigations, they discover a cover-up scheme involving a pharmaceutical company, revealing a plot that reaches much further than the ragtag team of bounty hunters could have realized.
+
+[Written by MAL Rewrite]",,"[{'mal_id': 1, 'type': 'anime', 'name': 'Action', 'url': 'https://myanimelist.net/anime/genre/1/Action'}, {'mal_id': 24, 'type': 'anime', 'name': 'Sci-Fi', 'url': 'https://myanimelist.net/anime/genre/24/Sci-Fi'}]","[{'mal_id': 50, 'type': 'anime', 'name': 'Adult Cast', 'url': 'https://myanimelist.net/anime/genre/50/Adult_Cast'}, {'mal_id': 29, 'type': 'anime', 'name': 'Space', 'url': 'https://myanimelist.net/anime/genre/29/Space'}]","[{'name': 'Netflix', 'url': 'https://www.netflix.com/'}]"
+2,6,https://myanimelist.net/anime/6/Trigun,https://www.youtube.com/watch?v=bJVyIXeUznY,Trigun,26,Finished Airing,False,24 min per ep,PG-13 - Teens 13 or older,8.22,333,248,"Vash the Stampede is the man with a $$60,000,000,000 bounty on his head. The reason: he's a merciless villain who lays waste to all those that oppose him and flattens entire cities for fun, garnering him the title ""The Humanoid Typhoon."" He leaves a trail of death and destruction wherever he goes, and anyone can count themselves dead if they so much as make eye contact—or so the rumors say. In actuality, Vash is a huge softie who claims to have never taken a life and avoids violence at all costs.
+
+With his crazy doughnut obsession and buffoonish attitude in tow, Vash traverses the wasteland of the planet Gunsmoke, all the while followed by two insurance agents, Meryl Stryfe and Milly Thompson, who attempt to minimize his impact on the public. But soon, their misadventures evolve into life-or-death situations as a group of legendary assassins are summoned to bring about suffering to the trio. Vash's agonizing past will be unraveled and his morality and principles pushed to the breaking point.
+
+[Written by MAL Rewrite]",1998.0,"[{'mal_id': 1, 'type': 'anime', 'name': 'Action', 'url': 'https://myanimelist.net/anime/genre/1/Action'}, {'mal_id': 2, 'type': 'anime', 'name': 'Adventure', 'url': 'https://myanimelist.net/anime/genre/2/Adventure'}, {'mal_id': 24, 'type': 'anime', 'name': 'Sci-Fi', 'url': 'https://myanimelist.net/anime/genre/24/Sci-Fi'}]","[{'mal_id': 50, 'type': 'anime', 'name': 'Adult Cast', 'url': 'https://myanimelist.net/anime/genre/50/Adult_Cast'}]","[{'name': 'Crunchyroll', 'url': 'http://www.crunchyroll.com/series-275669'}, {'name': 'Funimation', 'url': 'https://www.funimation.com/shows/trigun'}, {'name': 'Netflix', 'url': 'https://www.netflix.com/'}]"
+3,7,https://myanimelist.net/anime/7/Witch_Hunter_Robin,https://www.youtube.com/watch?v=7UkaILjPk8M,Witch Hunter Robin,26,Finished Airing,False,25 min per ep,PG-13 - Teens 13 or older,7.24,2943,1828,"Robin Sena is a powerful craft user drafted into the STNJ—a group of specialized hunters that fight deadly beings known as Witches. Though her fire power is great, she's got a lot to learn about her powers and working with her cool and aloof partner, Amon. But the truth about the Witches and herself will leave Robin on an entirely new path that she never expected!
+
+(Source: Funimation)",2002.0,"[{'mal_id': 1, 'type': 'anime', 'name': 'Action', 'url': 'https://myanimelist.net/anime/genre/1/Action'}, {'mal_id': 8, 'type': 'anime', 'name': 'Drama', 'url': 'https://myanimelist.net/anime/genre/8/Drama'}, {'mal_id': 7, 'type': 'anime', 'name': 'Mystery', 'url': 'https://myanimelist.net/anime/genre/7/Mystery'}, {'mal_id': 37, 'type': 'anime', 'name': 'Supernatural', 'url': 'https://myanimelist.net/anime/genre/37/Supernatural'}]","[{'mal_id': 39, 'type': 'anime', 'name': 'Detective', 'url': 'https://myanimelist.net/anime/genre/39/Detective'}]","[{'name': 'Crunchyroll', 'url': 'http://www.crunchyroll.com/series-283299'}, {'name': 'Funimation', 'url': 'https://www.funimation.com/shows/witch-hunter-robin'}]"
+4,8,https://myanimelist.net/anime/8/Bouken_Ou_Beet,,Bouken Ou Beet,52,Finished Airing,False,23 min per ep,PG - Children,6.93,4440,5250,"It is the dark century and the people are suffering under the rule of the devil, Vandel, who is able to manipulate monsters. The Vandel Busters are a group of people who hunt these devils, and among them, the Zenon Squad is known to be the strongest busters on the continent. A young boy, Beet, dreams of joining the Zenon Squad. However, one day, as a result of Beet's fault, the Zenon squad was defeated by the devil, Beltose. The five dying busters sacrificed their life power into their five weapons, Saiga. After giving their weapons to Beet, they passed away. Years have passed since then and the young Vandel Buster, Beet, begins his adventure to carry out the Zenon Squad's will to put an end to the dark century.",2004.0,"[{'mal_id': 2, 'type': 'anime', 'name': 'Adventure', 'url': 'https://myanimelist.net/anime/genre/2/Adventure'}, {'mal_id': 10, 'type': 'anime', 'name': 'Fantasy', 'url': 'https://myanimelist.net/anime/genre/10/Fantasy'}, {'mal_id': 37, 'type': 'anime', 'name': 'Supernatural', 'url': 'https://myanimelist.net/anime/genre/37/Supernatural'}]",[],[]
+5,15,https://myanimelist.net/anime/15/Eyeshield_21,,Eyeshield 21,145,Finished Airing,False,23 min per ep,PG-13 - Teens 13 or older,7.91,741,1279,"Shy, reserved, and small-statured, Deimon High School student Sena Kobayakawa is the perfect target for bullies. However, as a result of running errands throughout his life, Sena has become agile and developed a skill for avoiding crowds of people. After the cunning Youichi Hiruma—captain of the Deimon Devil Bats football team—witnesses Sena's rapid legs in motion, he coerces the timid boy into joining his squad.
+
+As Hiruma wants to conceal Sena's identity from other clubs, Sena is forced to hide under the visored helmet of ""Eyeshield 21,"" a mysterious running back wearing the number 21 jersey. The legendary Eyeshield 21 can supposedly run at the speed of light and has achieved remarkable feats in the United States during his time at the Notre Dame College.
+
+Accustomed to avoiding his problems in the past, Sena's specialty might just help him become the new secret weapon of the Deimon Devil Bats. As he interacts with his teammates, Sena gradually gains more self-confidence and forges valuable bonds along the way.
+
+[Written by MAL Rewrite]",2005.0,"[{'mal_id': 30, 'type': 'anime', 'name': 'Sports', 'url': 'https://myanimelist.net/anime/genre/30/Sports'}]","[{'mal_id': 77, 'type': 'anime', 'name': 'Team Sports', 'url': 'https://myanimelist.net/anime/genre/77/Team_Sports'}]","[{'name': 'Crunchyroll', 'url': 'http://www.crunchyroll.com/series-46844'}]"
+6,16,https://myanimelist.net/anime/16/Hachimitsu_to_Clover,https://www.youtube.com/watch?v=6TN4a0kZuXg,Hachimitsu to Clover,24,Finished Airing,False,23 min per ep,PG-13 - Teens 13 or older,8.0,616,881,"Yuuta Takemoto, a sophomore at an arts college, shares a cheap apartment with two seniors—the eccentric Shinobu Morita, who keeps failing to graduate due to his absenteeism, and the sensible Takumi Mayama, who acts as a proper senior to Takemoto, often looking out for him.
+
+Takemoto had not given much thought to his future until one fine spring day, when he meets the endearing Hagumi Hanamoto and falls in love at first sight. Incredibly gifted in the arts, Hagumi enrolls in Takemoto's university and soon befriends the popular pottery student Ayumi Yamada. Ayumi is already well acquainted with the three flatmates and secretly harbors deep feelings for one of them.
+
+Hachimitsu to Clover is a heartwarming tale of youth, love, soul-searching, and self-discovery, intricately woven through the complex relationships between five dear friends.
+
+[Written by MAL Rewrite]",2005.0,"[{'mal_id': 4, 'type': 'anime', 'name': 'Comedy', 'url': 'https://myanimelist.net/anime/genre/4/Comedy'}, {'mal_id': 8, 'type': 'anime', 'name': 'Drama', 'url': 'https://myanimelist.net/anime/genre/8/Drama'}, {'mal_id': 22, 'type': 'anime', 'name': 'Romance', 'url': 'https://myanimelist.net/anime/genre/22/Romance'}]","[{'mal_id': 50, 'type': 'anime', 'name': 'Adult Cast', 'url': 'https://myanimelist.net/anime/genre/50/Adult_Cast'}, {'mal_id': 64, 'type': 'anime', 'name': 'Love Polygon', 'url': 'https://myanimelist.net/anime/genre/64/Love_Polygon'}, {'mal_id': 80, 'type': 'anime', 'name': 'Visual Arts', 'url': 'https://myanimelist.net/anime/genre/80/Visual_Arts'}]","[{'name': 'Crunchyroll', 'url': 'http://www.crunchyroll.com/series-277968'}]"
+7,17,https://myanimelist.net/anime/17/Hungry_Heart__Wild_Striker,,Hungry Heart: Wild Striker,52,Finished Airing,False,23 min per ep,PG-13 - Teens 13 or older,7.55,1635,4327,"As the younger brother of Japanese soccer star Seisuke Kanou, Kyousuke was always expected to grow as a soccer player at the same pace his brother did—an expectation that proved too difficult to meet. Having fallen behind, he now lives in the shadow of his brother's success.
+
+Entering his freshman year at Jouyou Akanegaoka High School, Kyousuke vows never to play soccer again. However, Miki Tsujiwaki, the captain of the girls' soccer team, and Mori Kazuto, the manager of the boys' team, recognize Kyousuke's potential and want to see his return to the game for their own reasons.
+
+With an opportunity to play soccer again, Kyousuke must either remain steadfast in his decision to abandon the sport he once loved, or allow himself to reignite that flame to become the best striker in the world.
+
+[Written by MAL Rewrite]",2002.0,"[{'mal_id': 4, 'type': 'anime', 'name': 'Comedy', 'url': 'https://myanimelist.net/anime/genre/4/Comedy'}, {'mal_id': 36, 'type': 'anime', 'name': 'Slice of Life', 'url': 'https://myanimelist.net/anime/genre/36/Slice_of_Life'}, {'mal_id': 30, 'type': 'anime', 'name': 'Sports', 'url': 'https://myanimelist.net/anime/genre/30/Sports'}]","[{'mal_id': 77, 'type': 'anime', 'name': 'Team Sports', 'url': 'https://myanimelist.net/anime/genre/77/Team_Sports'}]",[]
+8,18,https://myanimelist.net/anime/18/Initial_D_Fourth_Stage,,Initial D Fourth Stage,24,Finished Airing,False,27 min per ep,PG-13 - Teens 13 or older,8.16,410,1271,"Takumi Fujiwara finally joins Ryousuke and Keisuke Takahashi to create ""Project D."" Their goal is twofold: Ryousuke wants to develop his ""High-Speed Street Racing Theory,"" while Keisuke and Takumi aim at improving their driving skills by facing powerful opponents on dangerous roads. The idea of Project D is to challenge street racing teams from other prefectures to improve both their uphill and downhill records. In order to attract the attention of the best racing teams, Ryousuke creates a dedicated website to announce the future battles of Project D and post the team's results.
+
+The fourth season of Initial D details the hardships and successes of the members of Project D as they try to become the best street racing team outside of Gunma Prefecture.
+
+[Written by MAL Rewrite]",2004.0,"[{'mal_id': 1, 'type': 'anime', 'name': 'Action', 'url': 'https://myanimelist.net/anime/genre/1/Action'}, {'mal_id': 8, 'type': 'anime', 'name': 'Drama', 'url': 'https://myanimelist.net/anime/genre/8/Drama'}]","[{'mal_id': 3, 'type': 'anime', 'name': 'Racing', 'url': 'https://myanimelist.net/anime/genre/3/Racing'}]","[{'name': 'Crunchyroll', 'url': 'http://www.crunchyroll.com/series-282741'}, {'name': 'Funimation', 'url': 'https://www.funimation.com/shows/initial-d'}, {'name': 'Shahid', 'url': 'https://shahid.mbc.net/en/series/Initial-D:-Fourth-Stage/series-908421'}]"
+9,19,https://myanimelist.net/anime/19/Monster,,Monster,74,Finished Airing,False,24 min per ep,R+ - Mild Nudity,8.88,24,136,"Dr. Kenzou Tenma, an elite neurosurgeon recently engaged to his hospital director's daughter, is well on his way to ascending the hospital hierarchy. That is until one night, a seemingly small event changes Dr. Tenma's life forever. While preparing to perform surgery on someone, he gets a call from the hospital director telling him to switch patients and instead perform life-saving brain surgery on a famous performer. His fellow doctors, fiancée, and the hospital director applaud his accomplishment; but because of the switch, a poor immigrant worker is dead, causing Dr. Tenma to have a crisis of conscience.
+
+So when a similar situation arises, Dr. Tenma stands his ground and chooses to perform surgery on the young boy Johan Liebert instead of the town's mayor. Unfortunately, this choice leads to serious ramifications for Dr. Tenma—losing his social standing being one of them. However, with the mysterious death of the director and two other doctors, Dr. Tenma's position is restored. With no evidence to convict him, he is released and goes on to attain the position of hospital director. 
+
+Nine years later when Dr. Tenma saves the life of a criminal, his past comes back to haunt him—once again, he comes face to face with the monster he operated on. He must now embark on a quest of pursuit to make amends for the havoc spread by the one he saved.
+
+[Written by MAL Rewrite]",2004.0,"[{'mal_id': 8, 'type': 'anime', 'name': 'Drama', 'url': 'https://myanimelist.net/anime/genre/8/Drama'}, {'mal_id': 7, 'type': 'anime', 'name': 'Mystery', 'url': 'https://myanimelist.net/anime/genre/7/Mystery'}, {'mal_id': 41, 'type': 'anime', 'name': 'Suspense', 'url': 'https://myanimelist.net/anime/genre/41/Suspense'}]","[{'mal_id': 50, 'type': 'anime', 'name': 'Adult Cast', 'url': 'https://myanimelist.net/anime/genre/50/Adult_Cast'}, {'mal_id': 40, 'type': 'anime', 'name': 'Psychological', 'url': 'https://myanimelist.net/anime/genre/40/Psychological'}]",[]

recommend.py

@@ -0,0 +1,60 @@
+import cohere
+import os
+import pinecone
+import pandas as pd 
+import numpy as np
+from dotenv import dotenv_values
+
+env_name = ".env"
+config = dotenv_values(env_name)
+
+co = cohere.Client(os.environ['cohere_api'])
+
+pinecone.init(
+    api_key= config['pinecone_api'],
+    environment= config['pinecone_env']
+)
+
+index = pinecone.Index(config['pinecone_index'])
+
+
+
+def embed_text(text):
+    embeddings = co.embed(
+        texts=[text],
+        model='embed-english-v3.0',
+        input_type='search_query'
+    )
+
+    return embeddings
+
+
+
+def vector_search(desc):
+
+    results = []
+
+    embeddings = co.embed(
+        texts=[desc],
+        model='embed-english-v3.0',
+        input_type='search_query'
+    )
+
+    res = index.query([embeddings.embeddings[0]], top_k=2, include_metadata=True)
+
+    for match in res['matches']:
+        results.append(match['metadata']['text'])
+
+    return results
+
+def get_anime(res):
+    data = pd.read_csv('mini_data.csv')
+
+    df = pd.DataFrame()
+
+    for desc in res:
+        anime = data[data['description'] == desc]
+
+        df = pd.concat([df, anime])
+
+    return df 

requirements.txt

@@ -0,0 +1,10 @@
+requests 
+pathlib 
+pandas
+python-dotenv
+langchain
+openai 
+docarray
+tiktoken 
+cohere
+pinecone-client