Update app.py

app.py

@@ -767,13 +767,13 @@ with gr.Blocks(
                 )
                 cuisine_dropdown = gr.Dropdown(
                     label="요리 종류",
-                    choices=["전체"] + sorted(set(r['Cuisine'] for r in find_related_restaurants("", 1000))),
+                    choices=["전체"],  # find.csv 파일 존재 시 동적으로 업데이트
                     value="전체",
                     scale=1
                 )
                 award_dropdown = gr.Dropdown(
                     label="미쉐린 등급",
-                    choices=["전체"] + sorted(set(r['Award'] for r in find_related_restaurants("", 1000))),
+                    choices=["전체"],  # find.csv 파일 존재 시 동적으로 업데이트
                     value="전체",
                     scale=1
                 )
@@ -781,28 +781,53 @@ with gr.Blocks(
 
             result_table = gr.Dataframe(
                 headers=["Name", "Address", "Location", "Price", "Cuisine", "Award", "Description"],
-                row_count=10,
-                col_count=(7, "fixed"),
-                overflow_row_behaviour="paginate"
+                wrap=True,
+                max_rows=10,
+                interactive=False
             )
 
+            def init_dropdowns():
+                try:
+                    with open('find.csv', 'r', encoding='utf-8') as f:
+                        reader = csv.DictReader(f)
+                        restaurants = list(reader)
+                        cuisines = ["전체"] + sorted(set(r['Cuisine'] for r in restaurants if r['Cuisine']))
+                        awards = ["전체"] + sorted(set(r['Award'] for r in restaurants if r['Award']))
+                        return cuisines, awards
+                except FileNotFoundError:
+                    print("Warning: find.csv file not found")
+                    return ["전체"], ["전체"]
+
             def search_restaurants(search_term, cuisine, award):
-                restaurants = find_related_restaurants("", 1000)  # 전체 데이터 가져오기
-                
-                filtered = []
-                for r in restaurants:
-                    if search_term.lower() in r['Name'].lower() or \
-                       search_term.lower() in r['Address'].lower() or \
-                       search_term.lower() in r['Description'].lower():
-                        if (cuisine == "전체" or r['Cuisine'] == cuisine) and \
-                           (award == "전체" or r['Award'] == award):
-                            filtered.append([
-                                r['Name'], r['Address'], r['Location'],
-                                r['Price'], r['Cuisine'], r['Award'],
-                                r['Description']
-                            ])
-                
-                return filtered
+                try:
+                    with open('find.csv', 'r', encoding='utf-8') as f:
+                        reader = csv.DictReader(f)
+                        restaurants = list(reader)
+                    
+                    filtered = []
+                    search_term = search_term.lower() if search_term else ""
+                    
+                    for r in restaurants:
+                        if search_term == "" or \
+                           search_term in r['Name'].lower() or \
+                           search_term in r['Address'].lower() or \
+                           search_term in r['Description'].lower():
+                            if (cuisine == "전체" or r['Cuisine'] == cuisine) and \
+                               (award == "전체" or r['Award'] == award):
+                                filtered.append([
+                                    r['Name'], r['Address'], r['Location'],
+                                    r['Price'], r['Cuisine'], r['Award'],
+                                    r['Description']
+                                ])
+                    
+                    return filtered[:10]  # 최대 10개 결과만 반환
+                except FileNotFoundError:
+                    return [["파일을 찾을 수 없습니다", "", "", "", "", "", "find.csv 파일을 확인해주세요"]]
+
+            # 드롭다운 초기화
+            cuisines, awards = init_dropdowns()
+            cuisine_dropdown.choices = cuisines
+            award_dropdown.choices = awards
 
             search_button.click(
                 search_restaurants,