app.py

# 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,load_tools
from langchain.tools import Tool
from langchain_experimental.tools.python.tool import PythonREPLTool
#from langchain_community.retrievers import WikipediaRetriever
#from langchain.tools.retriever import create_retriever_tool
#retriever = WikipediaRetriever(lang="ko",top_k_results=10)
#retriever_tool = create_retriever_tool(
    #retriever,
   # name="wiki_search",
   # description="위키백과에서 필요한 정보를 불러옵니다.결괴를 검증하여 사용하시오.",
#)
# ──────────────────────────────
# ✅ 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": "Bearer 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",
            "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(
    ["ddg-search", "requests_all", "llm-math"],
    llm=llm,allow_dangerous_tools=True
)

# ──────────────────────────────
# ✅ 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 "❌ 파일 없음"

# ──────────────────────────────
# ✅ 정확한 한국 시간 함수 (Asia/Seoul)
# ──────────────────────────────
import requests
from datetime import datetime
from zoneinfo import ZoneInfo

def time_now(_=""):
    try:
        # 정확한 UTC 시각을 외부 API에서 가져옴
        resp = requests.get("https://timeapi.io/api/Time/current/zone?timeZone=Asia/Seoul", timeout=5)
        if resp.status_code == 200:
            data = resp.json()
            dt = data["dateTime"].split(".")[0].replace("T", " ")
            return f"현재 시각: {dt} (Asia/Seoul, 서버 기준 NTP 동기화)"
        else:
            # API 실패 시 로컬 시스템 시각으로 대체
            tz = ZoneInfo("Asia/Seoul")
            now = datetime.now(tz)
            return f"현재 시각(로컬): {now.strftime('%Y-%m-%d %H:%M:%S')} (Asia/Seoul)"
    except Exception as e:
        tz = ZoneInfo("Asia/Seoul")
        now = datetime.now(tz)
        return f"현재 시각(백업): {now.strftime('%Y-%m-%d %H:%M:%S')} (Asia/Seoul, 오류: {e})"
# ──────────────────────────────
# ✅ 도구 등록
# ──────────────────────────────
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,
    handle_parsing_errors=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 · Math · Requests · Python REPL · File · Time
    """)
    chatbot = gr.Chatbot(
    label="PIXAL(Pimary Interactive X-ternal Assistant with multi Language) 1.0",
    show_copy_button=True,
    layout="panel"
    )
    ai1=gr.ChatInterface(
    chat,
    chatbot=chatbot,    
    )

if __name__ == "__main__":
   ai1.launch()