Create app.py

app.py

@@ -0,0 +1,148 @@
+# pixal_agent_full.py
+import os
+import datetime
+import gradio as gr
+import requests
+from typing import Optional, List
+from langchain.llms.base import LLM
+from langchain.agents import initialize_agent, AgentType
+from langchain.tools import Tool, load_tools
+from langchain.tools.python.tool import PythonREPLTool
+
+# ──────────────────────────────
+# ✅ GitHub Models LLM
+# ──────────────────────────────
+class GitHubModelLLM(LLM):
+    model: str = "openai/gpt-4.1"
+    endpoint: str = "https://models.github.ai/inference"
+    token: Optional[str] = None
+
+    @property
+    def _llm_type(self) -> str:
+        return "github_models_api"
+
+    def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
+        if not self.token:
+            raise ValueError("GitHub API token이 필요합니다.")
+
+        headers = {
+            "Authorization": f"Bearer {self.token}",
+            "Content-Type": "application/json",
+        }
+        body = {"model": self.model, "messages": [{"role": "user", "content": prompt}]}
+
+        resp = requests.post(f"{self.endpoint}/chat/completions", json=body, headers=headers)
+        if resp.status_code != 200:
+            raise ValueError(f"API 오류: {resp.status_code} - {resp.text}")
+        return resp.json()["choices"][0]["message"]["content"]
+
+# ──────────────────────────────
+# ✅ LLM 설정
+# ──────────────────────────────
+token = os.getenv("GITHUB_TOKEN") or os.getenv("token")
+if not token:
+    print("⚠️ GitHub Token이 필요합니다. 예: setx GITHUB_TOKEN your_token")
+
+llm = GitHubModelLLM(model="openai/gpt-4.1", token=token)
+
+# ──────────────────────────────
+# ✅ LangChain 기본 도구 불러오기
+# ──────────────────────────────
+tools = load_tools(
+    ["duckduckgo-search", "requests-all", "llm-math", "wikipedia", "youtube-search"],
+    llm=llm
+)
+
+# ──────────────────────────────
+# ✅ Python 실행 도구 (LangChain 내장)
+# ──────────────────────────────
+python_tool = PythonREPLTool()
+tools.append(Tool(name="python_repl", func=python_tool.run, description="Python 코드를 실행합니다."))
+
+# ──────────────────────────────
+# ✅ 파일 도구
+# ──────────────────────────────
+BASE_DIR = os.path.join(os.getcwd(), "pixal_files")
+os.makedirs(BASE_DIR, exist_ok=True)
+
+def file_write(data: str) -> str:
+    try:
+        name, content = data.split("\n", 1)
+        path = os.path.join(BASE_DIR, name.strip())
+        with open(path, "w", encoding="utf-8") as f:
+            f.write(content)
+        return f"✅ 파일 저장됨: {path}"
+    except Exception as e:
+        return f"⚠️ 파일 저장 오류: {e}"
+
+def file_read(filename: str) -> str:
+    path = os.path.join(BASE_DIR, filename.strip())
+    if not os.path.exists(path):
+        return "❌ 파일 없음"
+    with open(path, "r", encoding="utf-8") as f:
+        return f.read()
+
+def file_list(_="") -> str:
+    return "\n".join(os.listdir(BASE_DIR))
+
+def file_delete(filename: str) -> str:
+    path = os.path.join(BASE_DIR, filename.strip())
+    if os.path.exists(path):
+        os.remove(path)
+        return f"🗑️ 삭제 완료: {filename}"
+    return "❌ 파일 없음"
+
+# ──────────────────────────────
+# ✅ 시간 도구
+# ──────────────────────────────
+def time_now(_=""):
+    now = datetime.datetime.now(datetime.timezone(datetime.timedelta(hours=9)))
+    return f"현재 시각: {now.strftime('%Y-%m-%d %H:%M:%S')} (Asia/Seoul)"
+
+# ──────────────────────────────
+# ✅ 도구 등록
+# ──────────────────────────────
+tools.extend([
+    Tool(name="file_write", func=file_write, description="파일을 생성/수정합니다."),
+    Tool(name="file_read", func=file_read, description="파일 내용을 읽습니다."),
+    Tool(name="file_list", func=file_list, description="파일 목록을 표시합니다."),
+    Tool(name="file_delete", func=file_delete, description="파일을 삭제합니다."),
+    Tool(name="time_now", func=time_now, description="현재 시간을 반환합니다."),
+])
+
+# ──────────────────────────────
+# ✅ Agent 초기화
+# ─────────────────────���────────
+agent = initialize_agent(
+    tools,
+    llm,
+    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+    verbose=True
+)
+
+# ──────────────────────────────
+# ✅ Gradio UI
+# ──────────────────────────────
+def chat(message, history):
+    try:
+        response = agent.run(message)
+    except Exception as e:
+        response = f"⚠️ 오류: {e}"
+    history = history + [(message, response)]
+    return history, history
+
+with gr.Blocks(theme=gr.themes.Soft(), title="PIXAL Assistant (LangChain + GitHub LLM)") as demo:
+    gr.Markdown("""
+    ## 🤖 PIXAL Assistant  
+    **LangChain 기반 멀티툴 에이전트**  
+    🧰 DuckDuckGo · Wikipedia · YouTube · Math · Requests · Python REPL · File · Time
+    """)
+    chatbot = gr.Chatbot(label="PIXAL 대화", height=600, latex=True)
+    msg = gr.Textbox(label="메시지", placeholder="명령 또는 질문을 입력하세요...")
+    clear = gr.Button("초기화")
+
+    msg.submit(chat, [msg, chatbot], [chatbot, chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)