Update app.py

app.py

@@ -98,21 +98,19 @@ def on_select(space):
     try:
         summary = summarize_space(space)
         app_content = get_app_py_content(space['id'])
-        screenshot_info = take_screenshot(space['url'])
+        screenshot = take_screenshot(space['url'])
         info = f"선택된 Space: {space['name']} (ID: {space['id']})\n"
         info += f"Author: {space['author']}\n"
         info += f"Likes: {space['likes']}\n"
         info += f"URL: {space['url']}\n\n"
-        info += f"요약:\n{summary}\n\n"
-        info += f"Screenshot Info: {screenshot_info}"
-        return info, app_content
+        info += f"요약:\n{summary}"
+        return info, app_content, screenshot
     except Exception as e:
         print(f"Error in on_select: {str(e)}")
         print(traceback.format_exc())
-        return f"오류가 발생했습니다: {str(e)}", ""
+        return f"오류가 발생했습니다: {str(e)}", "", Image.new('RGB', (1, 1))
 
 def take_screenshot(url):
-    return f"Screenshot functionality is disabled. URL: {url}"
     try:
         chrome_options = Options()
         chrome_options.add_argument("--headless")
@@ -121,7 +119,6 @@ def take_screenshot(url):
         chrome_options.add_argument("--disable-gpu")
         chrome_options.add_argument("--window-size=1080,720")
         
-        # 컨테이너 환경에 맞는 ChromeDriver 경로 설정
         driver = webdriver.Chrome(options=chrome_options)
         
         driver.get(url)
@@ -133,7 +130,6 @@ def take_screenshot(url):
         print(f"Screenshot error: {str(e)}")
         return Image.new('RGB', (1, 1))
 
-
 def create_ui():
     try:
         spaces_list = get_most_liked_spaces()
@@ -162,16 +158,18 @@ def create_ui():
 
                 with gr.Column(scale=1):
                     info_output = gr.Textbox(label="Space 정보 및 요약", lines=20)
+                    screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=540)
                     app_py_content = gr.Code(language="python", label="app.py 내용")
 
                 for _, button, space in space_rows:
                     button.click(
                         lambda s=space: on_select(s),
                         inputs=[],
-                        outputs=[info_output, app_py_content]
+                        outputs=[info_output, app_py_content, screenshot_output]
                     )
 
 
+
         return demo
 
     except Exception as e: