Update app.py

app.py

@@ -567,6 +567,62 @@ def update_placeholder():
 def create_main_interface():
     """메인 인터페이스 생성 함수"""
     
+    #NEW - 검색 결과를 통합한 응답 생성 함수
+    def execute_search_and_generate(query, setting):
+        try:
+            # 검색 실행
+            url = "https://api.serphouse.com/serp/live"
+            payload = {
+                "data": {
+                    "q": query,
+                    "domain": "google.com", 
+                    "lang": "en",
+                    "device": "desktop",
+                    "serp_type": "web",
+                    "loc": "Alba,Texas,United States",
+                    "loc_id": "1026201",
+                    "verbatim": "0",
+                    "gfilter": "0", 
+                    "page": "1",
+                    "num_result": "5"  # 상위 5개 결과만 사용
+                }
+            }
+            headers = {
+                'Authorization': 'Bearer V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6',
+                'Content-Type': 'application/json'
+            }
+            
+            response = requests.post(url, headers=headers, json=payload)
+            results = response.json()
+            
+            # 검색 결과를 HTML로 변환
+            search_content = "<div class='search-summary'><h2>Search Results</h2>"
+            for result in results['results'][:5]:  # 상위 5개 결과만 사용
+                search_content += f"""
+                    <div class="search-item">
+                        <h3><a href="{result['url']}" target="_blank">{result['title']}</a></h3>
+                        <p>{result['snippet']}</p>
+                    </div>
+                """
+            search_content += "</div>"
+            
+            # 검색 결과를 포함한 프롬프트 생성
+            enhanced_prompt = f"""Based on these search results, create a visually appealing and informative response:
+            {search_content}
+            
+            Original query: {query}
+            
+            Create a comprehensive visual response that incorporates relevant information from the search results.
+            """
+            
+            # 기존 generation_code 함수 호출
+            return demo_instance.generation_code(enhanced_prompt, setting)
+            
+        except Exception as e:
+            print(f"Search error: {str(e)}")
+            # 검색 실패시 기본 generation_code로 폴백
+            return demo_instance.generation_code(query, setting)
+
     def execute_code(query: str):
         if not query or query.strip() == '':
             return None, gr.update(active_key="empty")
@@ -586,7 +642,39 @@ def create_main_interface():
     with open('app.css', 'r', encoding='utf-8') as f:
         custom_css = f.read()
 
-    demo = gr.Blocks(css=custom_css + """
+    #NEW - 검색 결과 스타일 추가
+    custom_css += """
+    .search-summary {
+        margin: 20px 0;
+        padding: 20px;
+        background: #f8f9fa;
+        border-radius: 10px;
+    }
+
+    .search-item {
+        margin-bottom: 15px;
+        padding: 15px;
+        border-left: 4px solid #007aff;
+        background: white;
+        border-radius: 4px;
+    }
+
+    .search-item h3 {
+        margin: 0 0 10px 0;
+        color: #1a0dab;
+    }
+
+    .search-item a {
+        color: inherit;
+        text-decoration: none;
+    }
+
+    .search-item p {
+        margin: 0;
+        color: #4d5156;
+        font-size: 14px;
+        line-height: 1.5;
+    }
 
         .empty-content {
             padding: 40px !important;
@@ -694,15 +782,18 @@ def create_main_interface():
     """, theme=theme)
 
 
-
     with demo:
         with gr.Tabs(elem_classes="main-tabs") as tabs:
             with gr.Tab("Visual AI Assistant", elem_id="mouse-tab", elem_classes="mouse-tab"):
-                # SystemPrompt 설정을 위한 State 추가
                 setting = gr.State({
                     "system": SystemPrompt,
                 })
                 
+                #NEW - 검색 모드 상태 추가
+                search_mode = gr.State(False)
+                
+
+                
                 with ms.Application() as app:
                     with antd.ConfigProvider():
 
@@ -921,6 +1012,33 @@ Use the "Generate" button for basic creation, "Enhance" button for prompt improv
                                         elem_classes="deploy-result"
                                     )
 
+                                    
+                # 버튼 이벤트 핸들러 수정
+                                    btn.click(
+                                        fn=lambda q, s, m: execute_search_and_generate(q, s) if m else demo_instance.generation_code(q, s),
+                                        inputs=[input, setting, search_mode],
+                                        outputs=[code_output, sandbox, state_tab, code_drawer]
+                                    ).then(
+                                        fn=update_placeholder,
+                                        inputs=[],
+                                        outputs=[input]
+                                    )
+
+                #NEW - 검색 모드 토글 버튼 추가
+                                    search_toggle = antd.Button(
+                                        "Toggle Search Mode",
+                                        type="default",
+                                        size="large",
+                                        elem_classes="search-toggle-btn"
+                                    )
+                            
+                                    search_toggle.click(
+                                        fn=lambda x: not x,
+                                        inputs=[search_mode],
+                                        outputs=[search_mode]
+                                    )
+
+                
                             # 우측 패널
                             with antd.Col(span=24, md=16):
                                 with ms.Div(elem_classes="right_panel"):