Update app.py

app.py

@@ -13,10 +13,14 @@ import threading
 import json
 import re
 
+# HuggingFace 관련 API 키 (스페이스 분석 용)
 HF_TOKEN = os.getenv("HF_TOKEN")
-hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
 hf_api = HfApi(token=HF_TOKEN)
 
+# Gemini 2.0 Thinking 모델 관련 API 키 및 클라이언트 (LLM 용)
+G_API_KEY = os.getenv("G_API_KEY")
+gemini_client = InferenceClient("Gemini-2.0-thinking", token=G_API_KEY)
+
 def get_headers():
     if not HF_TOKEN:
         raise ValueError("Hugging Face token not found in environment variables")
@@ -81,7 +85,7 @@ def summarize_code(app_content: str):
     ]
     
     try:
-        response = hf_client.chat_completion(messages, max_tokens=200, temperature=0.7)
+        response = gemini_client.chat_completion(messages, max_tokens=200, temperature=0.7)
         return response.choices[0].message.content
     except Exception as e:
         return f"요약 생성 중 오류 발생: {str(e)}"
@@ -102,7 +106,7 @@ def analyze_code(app_content: str):
     ]
     
     try:
-        response = hf_client.chat_completion(messages, max_tokens=1000, temperature=0.7)
+        response = gemini_client.chat_completion(messages, max_tokens=1000, temperature=0.7)
         return response.choices[0].message.content
     except Exception as e:
         return f"분석 생성 중 오류 발생: {str(e)}"
@@ -117,7 +121,7 @@ def explain_usage(app_content: str):
     ]
     
     try:
-        response = hf_client.chat_completion(messages, max_tokens=800, temperature=0.7)
+        response = gemini_client.chat_completion(messages, max_tokens=800, temperature=0.7)
         return response.choices[0].message.content
     except Exception as e:
         return f"사용법 설명 생성 중 오류 발생: {str(e)}"
@@ -195,7 +199,7 @@ def respond(
     messages.append({"role": "user", "content": message})
 
     response = ""
-    for message in hf_client.chat_completion(
+    for message in gemini_client.chat_completion(
         messages,
         max_tokens=max_tokens,
         stream=True,
@@ -288,7 +292,7 @@ def create_ui():
         """
 
         with gr.Blocks(theme="default", css=css) as demo:
-            gr.Markdown("# MOUSE Space Analysis", elem_classes="header-markdown")
+            gr.Markdown("# MOUSE: HF Space Deep-Research", elem_classes="header-markdown")
             
             with gr.Tabs() as tabs:
                 with gr.TabItem("분석"):