Update app.py

app.py

@@ -141,24 +141,36 @@ class AnimeRecommender:
     def NCF_Recommendation(self, a, b, c):
         anime_list = [a, b, c]
         anime_result = self.find_similar_animes_combined(anime_list, n=3)
-
-        result1 = anime_result.loc[3, 'name']
-        result2 = anime_result.loc[2, 'name']
-        result3 = anime_result.loc[1, 'name']
-
-        explain1 = anime_result.loc[3, 'sypnopsis']
-        explain2 = anime_result.loc[2, 'sypnopsis']
-        explain3 = anime_result.loc[1, 'sypnopsis']
-
-        url1 = self.get_anime_url(result1)
-        url2 = self.get_anime_url(result2)
-        url3 = self.get_anime_url(result3)
-
-        image1 = self.extract_image_url(url1)
-        image2 = self.extract_image_url(url2)
-        image3 = self.extract_image_url(url3)
-
-        return result1, explain1, image1, result2, explain2, image2, result3, explain3, image3
+        
+        results = []
+        explains = []
+        images = []
+    
+        for i in range(min(len(anime_result), 3)):
+            try:
+                result = anime_result.loc[anime_result.index[i], 'name']
+                explain = anime_result.loc[anime_result.index[i], 'sypnopsis']
+                url = self.get_anime_url(result)
+                image = self.extract_image_url(url)
+    
+                results.append(result)
+                explains.append(explain)
+                images.append(image)
+            except Exception as e:
+                print(f"오류 발생: {e}")
+                break
+    
+        # 부족한 부분을 빈 문자열로 채우기
+        while len(results) < 3:
+            results.append("")
+            explains.append("")
+            images.append("")
+    
+        return (
+            results[0], explains[0], images[0],
+            results[1], explains[1], images[1],
+            results[2], explains[2], images[2]
+        )
 
 # 파일 경로 설정
 rating_path = 'data/rating_complete.csv'
@@ -198,4 +210,4 @@ with gr.Blocks() as app:
         outputs=[output1, output2, img1, output3, output4, img2, output5, output6, img3]
     )
 
-app.launch(share=True)
+app.launch()