Update app.py

app.py

@@ -93,6 +93,7 @@ def get_app_py_content(space_id: str) -> str:
         
 def on_select(space):
     try:
+        print(f"Selected space: {space['name']}")
         summary = summarize_space(space)
         app_content = get_app_py_content(space['id'])
         info = f"선택된 Space: {space['name']} (ID: {space['id']})\n"
@@ -100,6 +101,7 @@ def on_select(space):
         info += f"Likes: {space['likes']}\n"
         info += f"URL: {space['url']}\n\n"
         info += f"요약:\n{summary}"
+        print(f"Returning URL: {space['url']}")
         return info, app_content, space['url']
     except Exception as e:
         print(f"Error in on_select: {str(e)}")
@@ -107,19 +109,28 @@ def on_select(space):
         return f"오류가 발생했습니다: {str(e)}", "", ""
 
 def update_screenshot(url, last_url):
+    print(f"Updating screenshot. Current URL: {url}, Last URL: {last_url}")
     if url and url != last_url:
         screenshot = take_screenshot(url)
+        print("Screenshot updated")
         return screenshot, url
+    print("No update needed")
     return gr.update(), last_url
         
 def take_screenshot(url):
     try:
+        print(f"Attempting to take screenshot of URL: {url}")
         client = Client("ginipick/selenium-screenshot-gradio")
         result = client.predict(url=url, api_name="/predict")
-        return Image.open(result)
+        print(f"Screenshot result: {result}")
+        if isinstance(result, str) and os.path.exists(result):
+            return Image.open(result)
+        else:
+            print(f"Invalid result from API: {result}")
+            return Image.new('RGB', (540, 360), color='lightgray')
     except Exception as e:
         print(f"Screenshot error: {str(e)}")
-        return Image.new('RGB', (1080, 720), color='lightgray')
+        return Image.new('RGB', (540, 360), color='lightgray')
             
 def create_ui():
     try:
@@ -153,7 +164,7 @@ def create_ui():
                     last_url_state = gr.State("")
                     
                     with gr.Row():
-                        screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=540)
+                        screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
                         refresh_button = gr.Button("🔄 새로고침", scale=0.1)
                     
                     app_py_content = gr.Code(language="python", label="메인 소스코드")