Upload 6 files

Dockerfile

@@ -0,0 +1,53 @@
+FROM python:3.12.3-slim as builder
+
+WORKDIR /app
+
+# Install system dependencies required for building
+RUN apt-get update && apt-get install -y \
+    protobuf-compiler \
+    gcc \
+    g++ \
+    cmake \
+    pkg-config \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install poetry
+RUN pip install poetry
+
+# Copy dependency files
+COPY pyproject.toml poetry.lock ./
+
+# Configure poetry and install dependencies
+RUN poetry config virtualenvs.create false \
+    && poetry install --no-interaction --no-ansi --no-root
+
+# Second stage
+FROM python:3.12.3-slim
+
+WORKDIR /app
+
+# Copy installed dependencies from builder
+COPY --from=builder /usr/local/lib/python3.12/site-packages/ /usr/local/lib/python3.12/site-packages/
+COPY --from=builder /usr/local/bin/ /usr/local/bin/
+
+# Install runtime system dependencies
+RUN apt-get update && apt-get install -y \
+    protobuf-compiler \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy application code
+COPY . .
+
+# Create non-root user
+RUN useradd -m appuser && \
+    chown -R appuser:appuser /app
+USER appuser
+
+EXPOSE 7860
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/ || exit 1
+
+CMD ["poetry", "run", "python", "app.py"]

app.py

@@ -0,0 +1,106 @@
+import gradio as gr
+from scripts.github_analyzer import GitHubAnalyzer
+from scripts.topic_list import TOPIC_LIST
+from scripts.error_handler import ErrorHandler
+
+analyzer = GitHubAnalyzer()
+error_handler = ErrorHandler()
+
+async def process_url(
+    url: str,
+    main_cat: str,
+    sub_cat: str,
+    use_gpu: bool
+) -> tuple[str, str, str]:
+    try:
+        if not all([url, main_cat, sub_cat]):
+            response = error_handler.handle_github_url_error(
+                url,
+                "Please select all categories"
+            )
+            return response.errors[0].message, "", ""
+
+        analyzer.set_device("cuda" if use_gpu else "cpu")
+        response = await analyzer.analyze_repository(url, main_cat, sub_cat)
+
+        if not response.success:
+            return response.errors[0].message, "", ""
+
+        readme_topics = " ".join([
+            f"#{topic['topic'].lower()} ({topic['score']:.2f})"
+            for topic in response.data["readme_topics"]
+        ])
+
+        code_topics = " ".join([
+            f"#{topic['topic'].lower()} ({topic['score']:.2f})"
+            for topic in response.data["code_topics"]
+        ])
+
+        dependencies = " ".join([
+            f"#{dep.lower()}"
+            for dep in response.data["dependencies"]
+        ])
+
+        return readme_topics, code_topics, dependencies
+
+    except Exception as e:
+        response = error_handler.handle_topic_analysis_error(str(e))
+        return response.errors[0].message, "", ""
+
+def create_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Enhanced GitHub Topic Generator")
+
+        with gr.Row():
+            url_input = gr.Textbox(
+                label="GitHub URL",
+                placeholder="Enter GitHub repository URL"
+            )
+
+        with gr.Row():
+            main_category = gr.Dropdown(
+                choices=[None] + list(TOPIC_LIST.keys()),
+                label="Main Category",
+                value=None
+            )
+            sub_category = gr.Dropdown(
+                choices=[],
+                label="Sub Category"
+            )
+
+        with gr.Row():
+            use_gpu = gr.Checkbox(
+                label="Use GPU (Check if you have CUDA-capable GPU)",
+                value=False
+            )
+
+        with gr.Row():
+            generate_btn = gr.Button("Generate Topics")
+
+        with gr.Row():
+            readme_topics = gr.Textbox(label="README Topics")
+            code_topics = gr.Textbox(label="Code Analysis Topics")
+            dependencies = gr.Textbox(label="Dependencies")
+
+        def update_sub_category(main_cat):
+            return gr.Dropdown(
+                choices=list(TOPIC_LIST[main_cat].keys()) if main_cat else []
+            )
+
+        main_category.change(
+            update_sub_category,
+            inputs=main_category,
+            outputs=sub_category
+        )
+
+        generate_btn.click(
+            process_url,
+            inputs=[url_input, main_category, sub_category, use_gpu],
+            outputs=[readme_topics, code_topics, dependencies]
+        )
+
+    return demo
+
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch(share=True)

docker-compose.yml

@@ -0,0 +1,24 @@
+version: '3.8'
+
+services:
+  web:
+    build: .
+    ports:
+      - "7860:7860"
+    environment:
+      - GRADIO_SERVER_NAME=0.0.0.0
+      - GRADIO_SERVER_PORT=7860
+    deploy:
+      resources:
+        limits:
+          memory: 4G
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:7860/"]
+      interval: 30s
+      timeout: 30s
+      retries: 3
+    logging:
+      driver: "json-file"
+      options:
+        max-size: "10m"
+        max-file: "3"

pyproject.toml

@@ -0,0 +1,35 @@
+[tool.poetry]
+name = "github-topic-generator"
+version = "0.1.2"
+description = "GitHub repository topic generator using AI"
+authors = ["Namgyu-Youn <[email protected]>"]
+license = "MIT"
+readme = "README.md"
+package-mode = false
+
+[tool.poetry.dependencies]
+python = ">=3.10,<3.13"
+gradio = "^5.9.1"
+torch = {version = "^2.5.1", source = "pytorch"}
+transformers = "^4.47.1"
+aiohttp = "^3.11.11"
+python-dotenv = "^1.0.1"
+ruff = "^0.8.5"
+tiktoken = "^0.5.2"
+protobuf = "^4.25.3"
+sentencepiece = "^0.1.99"
+tokenizers = ">=0.21,<0.22"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^7.4.0"
+pytest-cov = "^4.1.0"
+pytest-asyncio = "^0.21.0"
+
+[[tool.poetry.source]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cpu"
+priority = "explicit"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -0,0 +1,6 @@
+numpy
+torch
+transformers
+gradio
+aiohttp
+python-dotenv