Update app.py

app.py

@@ -18,12 +18,14 @@ except ImportError:
     OPENAI_AVAILABLE = False
     print("OpenAI library not available. Install with: pip install openai")
 
-try:
-    import anthropic
-    ANTHROPIC_AVAILABLE = True
-except ImportError:
-    ANTHROPIC_AVAILABLE = False
-    print("Anthropic library not available. Install with: pip install anthropic")
+# Anthropic 관련 코드 주석 처리
+# try:
+#     import anthropic
+#     ANTHROPIC_AVAILABLE = True
+# except ImportError:
+#     ANTHROPIC_AVAILABLE = False
+#     print("Anthropic library not available. Install with: pip install anthropic")
+ANTHROPIC_AVAILABLE = False
 
 try:
     import requests
@@ -111,7 +113,8 @@ def create_sample_workflow():
                 "data": {
                     "label": "AI Processing",
                     "template": {
-                        "model": {"value": "gpt-3.5-turbo"},
+                        "provider": {"value": "OpenAI"},  # OpenAI or VIDraft
+                        "model": {"value": "gpt-4.1-mini"},  # Fixed to gpt-4.1-mini for OpenAI
                         "temperature": {"value": 0.7},
                         "system_prompt": {"value": "You are a helpful assistant."}
                     }
@@ -136,9 +139,9 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
     import traceback
     
     # API 키 확인
-    friendli_token = os.getenv("FRIENDLI_TOKEN")
+    vidraft_token = os.getenv("FRIENDLI_TOKEN")  # VIDraft/Friendli token
     openai_key = os.getenv("OPENAI_API_KEY")
-    anthropic_key = os.getenv("ANTHROPIC_API_KEY")
+    # anthropic_key = os.getenv("ANTHROPIC_API_KEY")  # 주석 처리
     
     # OpenAI 라이브러리 확인
     try:
@@ -148,13 +151,14 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
         openai_available = False
         print("OpenAI library not available")
     
-    # Anthropic 라이브러리 확인
-    try:
-        import anthropic
-        anthropic_available = True
-    except ImportError:
-        anthropic_available = False
-        print("Anthropic library not available")
+    # Anthropic 라이브러리 확인 - 주석 처리
+    # try:
+    #     import anthropic
+    #     anthropic_available = True
+    # except ImportError:
+    #     anthropic_available = False
+    #     print("Anthropic library not available")
+    anthropic_available = False
     
     results = {}
     nodes = workflow_data.get("nodes", [])
@@ -181,9 +185,23 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                 # LLM 노드 처리
                 template = node_data.get("template", {})
                 
+                # 프로바이더 정보 추출 - VIDraft 또는 OpenAI만 허용
+                provider_info = template.get("provider", {})
+                provider = provider_info.get("value", "OpenAI") if isinstance(provider_info, dict) else "OpenAI"
+                
+                # provider가 VIDraft 또는 OpenAI가 아닌 경우 OpenAI로 기본 설정
+                if provider not in ["VIDraft", "OpenAI"]:
+                    provider = "OpenAI"
+                
                 # 모델 정보 추출
-                model_info = template.get("model", {})
-                model = model_info.get("value", "gpt-3.5-turbo") if isinstance(model_info, dict) else "gpt-3.5-turbo"
+                if provider == "OpenAI":
+                    # OpenAI는 gpt-4.1-mini로 고정
+                    model = "gpt-4.1-mini"
+                elif provider == "VIDraft":
+                    # VIDraft는 Gemma-3-r1984-27B로 고정
+                    model = "Gemma-3-r1984-27B"
+                else:
+                    model = "gpt-4.1-mini"  # 기본값
                 
                 # 온도 정보 추출
                 temp_info = template.get("temperature", {})
@@ -193,10 +211,6 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                 prompt_info = template.get("system_prompt", {})
                 system_prompt = prompt_info.get("value", "") if isinstance(prompt_info, dict) else ""
                 
-                # 프로바이더 정보 추출
-                provider_info = template.get("provider", {})
-                provider = provider_info.get("value", "OpenAI") if isinstance(provider_info, dict) else "OpenAI"
-                
                 # 입력 텍스트 찾기
                 input_text = ""
                 for edge in edges:
@@ -217,7 +231,7 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                         messages.append({"role": "user", "content": input_text})
                         
                         response = client.chat.completions.create(
-                            model=model,
+                            model="gpt-4.1-mini",  # 고정된 모델명
                             messages=messages,
                             temperature=temperature,
                             max_tokens=1000
@@ -228,42 +242,49 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                     except Exception as e:
                         results[node_id] = f"[OpenAI Error: {str(e)}]"
                 
-                elif provider == "Anthropic" and anthropic_key and anthropic_available:
-                    try:
-                        client = anthropic.Anthropic(api_key=anthropic_key)
-                        
-                        message = client.messages.create(
-                            model="claude-3-haiku-20240307",
-                            max_tokens=1000,
-                            temperature=temperature,
-                            system=system_prompt if system_prompt else None,
-                            messages=[{"role": "user", "content": input_text}]
-                        )
-                        
-                        results[node_id] = message.content[0].text
-                        
-                    except Exception as e:
-                        results[node_id] = f"[Anthropic Error: {str(e)}]"
+                # Anthropic 관련 코드 주석 처리
+                # elif provider == "Anthropic" and anthropic_key and anthropic_available:
+                #     try:
+                #         client = anthropic.Anthropic(api_key=anthropic_key)
+                #         
+                #         message = client.messages.create(
+                #             model="claude-3-haiku-20240307",
+                #             max_tokens=1000,
+                #             temperature=temperature,
+                #             system=system_prompt if system_prompt else None,
+                #             messages=[{"role": "user", "content": input_text}]
+                #         )
+                #         
+                #         results[node_id] = message.content[0].text
+                #         
+                #     except Exception as e:
+                #         results[node_id] = f"[Anthropic Error: {str(e)}]"
                 
-                elif provider == "Friendly" and friendli_token:
+                elif provider == "VIDraft" and vidraft_token:
                     try:
                         import requests
                         
                         headers = {
-                            "Authorization": f"Bearer {friendli_token}",
+                            "Authorization": f"Bearer {vidraft_token}",
                             "Content-Type": "application/json"
                         }
                         
+                        # 메시지 구성
+                        messages = []
+                        if system_prompt:
+                            messages.append({"role": "system", "content": system_prompt})
+                        messages.append({"role": "user", "content": input_text})
+                        
                         payload = {
-                            "model": "dep89a2fld32mcm",
-                            "messages": [
-                                {"role": "system", "content": system_prompt} if system_prompt else {"role": "system", "content": "You are a helpful assistant."},
-                                {"role": "user", "content": input_text}
-                            ],
-                            "max_tokens": 1000,
-                            "temperature": temperature
+                            "model": "dep89a2fld32mcm",  # VIDraft 모델 ID
+                            "messages": messages,
+                            "max_tokens": 16384,
+                            "temperature": temperature,
+                            "top_p": 0.8,
+                            "stream": False  # 동기 실행을 위해 False로 설정
                         }
                         
+                        # VIDraft API endpoint
                         response = requests.post(
                             "https://api.friendli.ai/dedicated/v1/chat/completions",
                             headers=headers,
@@ -275,14 +296,14 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                             response_json = response.json()
                             results[node_id] = response_json["choices"][0]["message"]["content"]
                         else:
-                            results[node_id] = f"[Friendly API Error: {response.status_code}]"
+                            results[node_id] = f"[VIDraft API Error: {response.status_code} - {response.text}]"
                             
                     except Exception as e:
-                        results[node_id] = f"[Friendly Error: {str(e)}]"
+                        results[node_id] = f"[VIDraft Error: {str(e)}]"
                 
                 else:
                     # API 키가 없는 경우 시뮬레이션
-                    results[node_id] = f"[Simulated AI Response to: {input_text[:50]}...]"
+                    results[node_id] = f"[Simulated {provider} Response to: {input_text[:50]}...]"
             
             elif node_type in ["ChatOutput", "textOutput", "Output"]:
                 # 출력 노드는 연결된 노드의 결과를 가져옴
@@ -335,7 +356,7 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
     with gr.Column(elem_classes=["main-container"]):
         gr.Markdown("# 🐭 MOUSE Workflow")
         gr.Markdown("**Visual Workflow Builder with Interactive UI Execution**")
-        gr.HTML('<p class="powered-by">@Powered by VIDraft & Huggingface Gradio</p>')
+        gr.HTML('<p class="powered-by">@Powered by VIDraft & Huggingface gradio</p>')
         
         gr.HTML(
             """
@@ -349,15 +370,17 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         with gr.Accordion("🔌 API Status", open=False):
             gr.Markdown(f"""
             **Available APIs:**
-            - FRIENDLI_TOKEN: {'✅ Connected' if os.getenv("FRIENDLI_TOKEN") else '❌ Not found'}
+            - FRIENDLI_TOKEN (VIDraft): {'✅ Connected' if os.getenv("FRIENDLI_TOKEN") else '❌ Not found'}
             - OPENAI_API_KEY: {'✅ Connected' if os.getenv("OPENAI_API_KEY") else '❌ Not found'}
-            - ANTHROPIC_API_KEY: {'✅ Connected' if os.getenv("ANTHROPIC_API_KEY") else '❌ Not found'}
             
             **Libraries:**
             - OpenAI: {'✅ Installed' if OPENAI_AVAILABLE else '❌ Not installed'}
-            - Anthropic: {'✅ Installed' if ANTHROPIC_AVAILABLE else '❌ Not installed'}
             - Requests: {'✅ Installed' if REQUESTS_AVAILABLE else '❌ Not installed'}
             
+            **Available Models:**
+            - OpenAI: gpt-4.1-mini (fixed)
+            - VIDraft: Gemma-3-r1984-27B (model ID: dep89a2fld32mcm)
+            
             *Note: Without API keys, the UI will simulate AI responses.*
             """)
         
@@ -517,16 +540,14 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                     log += f"Inputs provided: {len(inputs_dict)}\n"
                     
                     # API 상태 확인
-                    friendli_token = os.getenv("FRIENDLI_TOKEN")
+                    vidraft_token = os.getenv("FRIENDLI_TOKEN")
                     openai_key = os.getenv("OPENAI_API_KEY")
-                    anthropic_key = os.getenv("ANTHROPIC_API_KEY")
                     
                     log += "\nAPI Status:\n"
-                    log += f"- FRIENDLI_TOKEN: {'✅ Found' if friendli_token else '❌ Not found'}\n"
+                    log += f"- FRIENDLI_TOKEN (VIDraft): {'✅ Found' if vidraft_token else '❌ Not found'}\n"
                     log += f"- OPENAI_API_KEY: {'✅ Found' if openai_key else '❌ Not found'}\n"
-                    log += f"- ANTHROPIC_API_KEY: {'✅ Found' if anthropic_key else '❌ Not found'}\n"
                     
-                    if not friendli_token and not openai_key and not anthropic_key:
+                    if not vidraft_token and not openai_key:
                         log += "\n⚠️ No API keys found. Results will be simulated.\n"
                         log += "To get real AI responses, set API keys in environment variables.\n"