Update app-backup4.py

app-backup4.py

@@ -4,6 +4,12 @@ import os
 import requests
 from typing import List, Dict, Union
 import traceback
+from PIL import Image
+from io import BytesIO
+import asyncio
+from gradio_client import Client
+import time
+import threading
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
@@ -76,14 +82,18 @@ def get_app_py_content(space_id: str) -> str:
     try:
         response = requests.get(app_py_url, headers=get_headers())
         if response.status_code == 200:
-            return response.text
+            content = response.text
+            if len(content) > 500:  # 내용을 500자로 제한합니다
+                content = content[:497] + "..."
+            return content
         else:
             return f"app.py file not found or inaccessible for space: {space_id}"
     except requests.RequestException:
         return f"Error fetching app.py content for space: {space_id}"
-
+        
 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"
@@ -91,50 +101,132 @@ def on_select(space):
         info += f"Likes: {space['likes']}\n"
         info += f"URL: {space['url']}\n\n"
         info += f"요약:\n{summary}"
-        return info, app_content
+        print(f"Returning URL: {space['url']}")
+        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)}", ""
+        return f"오류가 발생했습니다: {str(e)}", "", ""
+
+def update_screenshot(url, last_url, force_update=False):
+    print(f"Updating screenshot. Current URL: {url}, Last URL: {last_url}, Force update: {force_update}")
+    if url and (url != last_url or force_update):
+
+        screenshot = take_screenshot(url)
+        print("Screenshot updated")
+        return screenshot, url
+    print("No update needed")
+    return gr.update(), last_url
 
+def refresh_screenshot(url, last_url):
+    print(f"Refresh button clicked. URL: {url}, Last URL: {last_url}")
+    # 항상 강제로 업데이트
+    return update_screenshot(url, last_url, force_update=True)
+
+
+def take_screenshot(url):
+    try:
+        print(f"Taking screenshot of URL: {url}")
+        client = Client("ginipick/selenium-screenshot-gradio")
+        result = client.predict(url=url, api_name="/predict")
+        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', (600, 360), color='lightgray')
+    except Exception as e:
+        print(f"Screenshot error: {str(e)}")
+        return Image.new('RGB', (600, 360), color='lightgray')
+            
 def create_ui():
-    spaces_list = get_most_liked_spaces()
-    print(f"Type of spaces_list: {type(spaces_list)}")
-    formatted_spaces = format_spaces(spaces_list)
-    print(f"Total spaces loaded: {len(formatted_spaces)}")
-
-    css = """
-    footer {visibility: hidden;}
-    .minimal-button {min-width: 30px !important; height: 25px !important; line-height: 1 !important; font-size: 12px !important; padding: 2px 5px !important;}
-    .space-row {margin-bottom: 5px !important;}
-    """
-
-    with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
-        gr.Markdown("# Hugging Face Most Liked Spaces")
-        
-        with gr.Row():
-            with gr.Column(scale=1):
-                space_rows = []
-                for space in formatted_spaces:
-                    with gr.Row(elem_classes="space-row") as space_row:
-                        with gr.Column():
-                            gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
-                            button = gr.Button("클릭", elem_classes="minimal-button")
-                    space_rows.append((space_row, button, space))
+    try:
+        spaces_list = get_most_liked_spaces()
+        print(f"Type of spaces_list: {type(spaces_list)}")
+        formatted_spaces = format_spaces(spaces_list)
+        print(f"Total spaces loaded: {len(formatted_spaces)}")
+
+        css = """
+        footer {visibility: hidden;}
+        .minimal-button {min-width: 30px !important; height: 25px !important; line-height: 1 !important; font-size: 12px !important; padding: 2px 5px !important;}
+        .space-row {margin-bottom: 5px !important;}
+        #refresh-button {
+            width: 100% !important;
+            margin-top: 5px !important;
+        }
+        """
+
+        with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
+            gr.Markdown("# 300: HuggingFace Most Liked Spaces")
             
-            with gr.Column(scale=1):
-                info_output = gr.Textbox(label="Space 정보 및 요약", lines=16)
-                app_py_content = gr.Code(language="python", label="app.py 내용")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    space_rows = []
+                    for space in formatted_spaces:
+                        with gr.Row(elem_classes="space-row") as space_row:
+                            with gr.Column():
+                                gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
+                                button = gr.Button("클릭", elem_classes="minimal-button")
+                        space_rows.append((space_row, button, space))
+
+                with gr.Column(scale=1):
+                    info_output = gr.Textbox(label="Space 정보 및 요약", lines=20)
+                    url_state = gr.State("")
+                    last_url_state = gr.State("")
+                    
+                    screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
+                    refresh_button = gr.Button("🔄 새로고침", elem_id="refresh-button")
+                    
+                    app_py_content = gr.Code(language="python", label="메인 소스코드")
+                    update_trigger = gr.Button("Update Screenshot", visible=False)
+
+                for _, button, space in space_rows:
+                    button.click(
+                        lambda s=space: on_select(s),
+                        inputs=[],
+                        outputs=[info_output, app_py_content, url_state]
+                    ).then(
+                        update_screenshot,
+                        inputs=[url_state, last_url_state],
+                        outputs=[screenshot_output, last_url_state]
+                    )
+
+                def refresh_screenshot(url, last_url):
+                    print(f"Refresh button clicked. URL: {url}, Last URL: {last_url}")
+                    return update_screenshot(url, last_url)
+
+                refresh_button.click(
+                    refresh_screenshot,
+                    inputs=[url_state, last_url_state],
+                    outputs=[screenshot_output, last_url_state]
+                )
+
+
+                update_trigger.click(
+                    update_screenshot,
+                    inputs=[url_state, last_url_state],
+                    outputs=[screenshot_output, last_url_state]
+                )
 
-        for _, button, space in space_rows:
-            button.click(
-                lambda s=space: on_select(s),
-                inputs=[],
-                outputs=[info_output, app_py_content]
-            )
+        # Start a background thread to trigger updates
+        def trigger_updates():
+            while True:
+                time.sleep(5)
+                update_trigger.click()
 
-    return demo
+        threading.Thread(target=trigger_updates, daemon=True).start()
+
+        return demo
+
+    except Exception as e:
+        print(f"Error in create_ui: {str(e)}")
+        print(traceback.format_exc())
+        raise
 
 if __name__ == "__main__":
-    demo = create_ui()
-    demo.launch()
+    try:
+        demo = create_ui()
+        demo.launch()
+    except Exception as e:
+        print(f"Error in main: {str(e)}")
+        print(traceback.format_exc())