OhST commited on
Commit
2833827
·
verified ·
1 Parent(s): c81dfa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -30
app.py CHANGED
@@ -141,36 +141,25 @@ class AnimeRecommender:
141
  def NCF_Recommendation(self, a, b, c):
142
  anime_list = [a, b, c]
143
  anime_result = self.find_similar_animes_combined(anime_list, n=3)
144
-
145
- results = []
146
- explains = []
147
- images = []
148
-
149
- for i in range(min(len(anime_result), 3)):
150
- try:
151
- result = anime_result.loc[anime_result.index[i], 'name']
152
- explain = anime_result.loc[anime_result.index[i], 'sypnopsis']
153
- url = self.get_anime_url(result)
154
- image = self.extract_image_url(url)
155
-
156
- results.append(result)
157
- explains.append(explain)
158
- images.append(image)
159
- except Exception as e:
160
- print(f"오류 발생: {e}")
161
- break
162
-
163
- # 부족한 부분을 빈 문자열로 채우기
164
- while len(results) < 3:
165
- results.append("")
166
- explains.append("")
167
- images.append("")
168
-
169
- return (
170
- results[0], explains[0], images[0],
171
- results[1], explains[1], images[1],
172
- results[2], explains[2], images[2]
173
- )
174
 
175
  # 파일 경로 설정
176
  rating_path = 'data/rating_complete.csv'
 
141
  def NCF_Recommendation(self, a, b, c):
142
  anime_list = [a, b, c]
143
  anime_result = self.find_similar_animes_combined(anime_list, n=3)
144
+
145
+ result1 = anime_result.loc[3, 'name']
146
+ result2 = anime_result.loc[2, 'name']
147
+ result3 = anime_result.loc[1, 'name']
148
+
149
+ explain1 = anime_result.loc[3, 'sypnopsis']
150
+ explain2 = anime_result.loc[2, 'sypnopsis']
151
+ explain3 = anime_result.loc[1, 'sypnopsis']
152
+
153
+ url1 = self.get_anime_url(result1)
154
+ url2 = self.get_anime_url(result2)
155
+ url3 = self.get_anime_url(result3)
156
+
157
+ image1 = self.extract_image_url(url1)
158
+ image2 = self.extract_image_url(url2)
159
+ image3 = self.extract_image_url(url3)
160
+
161
+ return result1, explain1, image1, result2, explain2, image2, result3, explain3, image3
162
+
 
 
 
 
 
 
 
 
 
 
 
163
 
164
  # 파일 경로 설정
165
  rating_path = 'data/rating_complete.csv'