Update app.py

app.py

@@ -93,17 +93,21 @@ def on_select(space):
     try:
         summary = summarize_space(space)
         app_content = get_app_py_content(space['id'])
-        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}"
-        return info, app_content, screenshot
+        return info, app_content, space['url']
     except Exception as e:
         print(f"Error in on_select: {str(e)}")
         print(traceback.format_exc())
-        return f"오류가 발생했습니다: {str(e)}", "", Image.new('RGB', (1080, 720), color='lightgray')
+        return f"오류가 발생했습니다: {str(e)}", "", ""
+
+def update_screenshot(url):
+    if url:
+        return take_screenshot(url)
+    return None
         
 def take_screenshot(url):
     try:
@@ -142,16 +146,18 @@ def create_ui():
 
                 with gr.Column(scale=1):
                     info_output = gr.Textbox(label="Space 정보 및 요약", lines=20)
-                    screenshot_output = gr.Image(type="pil", label="Space 스크린샷", height=360, width=600)
-                    app_py_content = gr.Code(language="python", label="app.py 내용")
+                    screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
+                    app_py_content = gr.Code(language="python", label="메인 소스코드")
+                    url_state = gr.State("")
 
                 for _, button, space in space_rows:
                     button.click(
                         lambda s=space: on_select(s),
-                            inputs=[],
-                            outputs=[info_output, app_py_content, screenshot_output]
+                        inputs=[],
+                        outputs=[info_output, app_py_content, url_state]
                     )
 
+                demo.load(update_screenshot, inputs=[url_state], outputs=[screenshot_output], every=5)
 
         return demo