openfree commited on
Commit
72ef525
·
verified ·
1 Parent(s): e36fb77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -10
app.py CHANGED
@@ -586,7 +586,7 @@ def create_main_interface():
586
  "verbatim": "0",
587
  "gfilter": "0",
588
  "page": "1",
589
- "num_result": "5" # 상위 5개 결과만 사용
590
  }
591
  }
592
  headers = {
@@ -597,15 +597,24 @@ def create_main_interface():
597
  response = requests.post(url, headers=headers, json=payload)
598
  results = response.json()
599
 
600
- # 검색 결과를 HTML로 변환
601
  search_content = "<div class='search-summary'><h2>Search Results</h2>"
602
- for result in results['results'][:5]: # 상위 5개 결과만 사용
603
- search_content += f"""
604
- <div class="search-item">
605
- <h3><a href="{result['url']}" target="_blank">{result['title']}</a></h3>
606
- <p>{result['snippet']}</p>
607
- </div>
608
- """
 
 
 
 
 
 
 
 
 
609
  search_content += "</div>"
610
 
611
  # 검색 결과를 포함한 프롬프트 생성
@@ -622,7 +631,6 @@ def create_main_interface():
622
  final_result = result
623
  return final_result
624
 
625
-
626
  except Exception as e:
627
  print(f"Search error: {str(e)}")
628
  return [
@@ -643,6 +651,8 @@ def create_main_interface():
643
  except Exception as e:
644
  print(f"Generation error: {str(e)}")
645
  return ["", None, gr.update(active_key="error"), gr.update(open=False)]
 
 
646
 
647
  def execute_code(query: str):
648
  if not query or query.strip() == '':
 
586
  "verbatim": "0",
587
  "gfilter": "0",
588
  "page": "1",
589
+ "num_result": "5"
590
  }
591
  }
592
  headers = {
 
597
  response = requests.post(url, headers=headers, json=payload)
598
  results = response.json()
599
 
600
+ # 검색 결과를 HTML로 변환 - 수정된 부분
601
  search_content = "<div class='search-summary'><h2>Search Results</h2>"
602
+ if 'results' in results and isinstance(results['results'], list):
603
+ search_results = results['results']
604
+ if len(search_results) > 5:
605
+ search_results = search_results[:5]
606
+
607
+ for result in search_results:
608
+ title = result.get('title', 'No Title')
609
+ url = result.get('url', '#')
610
+ snippet = result.get('snippet', 'No description available')
611
+
612
+ search_content += f"""
613
+ <div class="search-item">
614
+ <h3><a href="{url}" target="_blank">{title}</a></h3>
615
+ <p>{snippet}</p>
616
+ </div>
617
+ """
618
  search_content += "</div>"
619
 
620
  # 검색 결과를 포함한 프롬프트 생성
 
631
  final_result = result
632
  return final_result
633
 
 
634
  except Exception as e:
635
  print(f"Search error: {str(e)}")
636
  return [
 
651
  except Exception as e:
652
  print(f"Generation error: {str(e)}")
653
  return ["", None, gr.update(active_key="error"), gr.update(open=False)]
654
+
655
+
656
 
657
  def execute_code(query: str):
658
  if not query or query.strip() == '':