refactor: refact app.py

webui/app.py → app.py

@@ -8,13 +8,13 @@ import gradio as gr
 
 from gradio_i18n import Translate, gettext as _
 
-from base import GraphGenParams
-from test_api import test_api_connection
-from cache_utils import setup_workspace, cleanup_workspace
-from count_tokens import count_tokens
+from webui.base import GraphGenParams
+from webui.test_api import test_api_connection
+from webui.cache_utils import setup_workspace, cleanup_workspace
+from webui.count_tokens import count_tokens
 
 # pylint: disable=wrong-import-position
-root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+root_dir = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(root_dir)
 
 from graphgen.graphgen import GraphGen
@@ -226,7 +226,7 @@ def run_graphgen(params, progress=gr.Progress()):
 with (gr.Blocks(title="GraphGen Demo", theme=gr.themes.Glass(),
                css=css) as demo):
     # Header
-    gr.Image(value=os.path.join(root_dir, 'resources', 'images', 'logo.png'),
+    gr.Image(value="https://github.com/open-sciencelab/GraphGen/blob/main/resources/images/logo.png?raw=true",
              label="GraphGen Banner",
              elem_id="banner",
              interactive=False,
@@ -256,7 +256,7 @@ with (gr.Blocks(title="GraphGen Demo", theme=gr.themes.Glass(),
         <a href="https://github.com/open-sciencelab/GraphGen/issues/10">
             <img src="https://img.shields.io/github/stars/open-sciencelab/GraphGen?style=social" alt="GitHub Stars">
         </a>
-        <a href="https://github.com/open-sciencelab/GraphGen/tree/main/resources">
+        <a href="https://arxiv.org/abs/2505.20416">
             <img src="https://img.shields.io/badge/arXiv-pdf-yellow" alt="arXiv">
         </a>
     </div>