Private-AI

Running

App Files Files Community

seawolf2357 commited on 23 days ago

Commit

c88aa2b

verified ·

1 Parent(s): 637b6bc

Update app.py

Browse files

Files changed (1) hide show

app.py +143 -7

app.py CHANGED Viewed

@@ -546,6 +546,7 @@ HTML_CONTENT = """<!DOCTYPE html>
         let animationFrame;
         let audioContext, analyser, audioSource;
         let dataChannel = null;
         // Web search toggle functionality
         searchToggle.addEventListener('click', () => {
@@ -575,16 +576,57 @@ HTML_CONTENT = """<!DOCTYPE html>
         sendButton.addEventListener('click', sendTextMessage);
-        function sendTextMessage() {
             const message = textInput.value.trim();
-            if (!message || !dataChannel || dataChannel.readyState !== 'open') return;
             addMessage('user', message);
-            dataChannel.send(JSON.stringify({
-                type: 'text_message',
-                content: message
-            }));
             textInput.value = '';
         }
         function updateStatus(state) {
@@ -592,12 +634,14 @@ HTML_CONTENT = """<!DOCTYPE html>
             if (state === 'connected') {
                 statusText.textContent = '연결됨';
                 sendButton.style.display = 'block';
             } else if (state === 'connecting') {
                 statusText.textContent = '연결 중...';
                 sendButton.style.display = 'none';
             } else {
                 statusText.textContent = '연결 대기 중';
-                sendButton.style.display = 'none';
             }
         }
         function updateButtonState() {
@@ -885,11 +929,80 @@ print(f"Search client initialized: {search_client is not None}, API key present:
 # Store connection settings
 connection_settings = {}
 def update_chatbot(chatbot: list[dict], response: ResponseAudioTranscriptDoneEvent):
     chatbot.append({"role": "assistant", "content": response.transcript})
     return chatbot
 class OpenAIHandler(AsyncStreamHandler):
     def __init__(self, web_search_enabled: bool = False, target_language: str = "",
                  system_prompt: str = "", webrtc_id: str = None) -> None:
@@ -1231,6 +1344,29 @@ async def custom_offer(request: Request):
     return response
 @app.post("/text_message/{webrtc_id}")
 async def receive_text_message(webrtc_id: str, request: Request):
     """Receive text message from client"""

         let animationFrame;
         let audioContext, analyser, audioSource;
         let dataChannel = null;
+        let isVoiceActive = false;
         // Web search toggle functionality
         searchToggle.addEventListener('click', () => {
         sendButton.addEventListener('click', sendTextMessage);
+        async function sendTextMessage() {
             const message = textInput.value.trim();
+            if (!message) return;
+            // Add user message to chat
             addMessage('user', message);
             textInput.value = '';
+            // Show sending indicator
+            const typingIndicator = document.createElement('div');
+            typingIndicator.classList.add('message', 'assistant');
+            typingIndicator.textContent = '입력 중...';
+            typingIndicator.id = 'typing-indicator';
+            chatMessages.appendChild(typingIndicator);
+            chatMessages.scrollTop = chatMessages.scrollHeight;
+            try {
+                // Send to text chat endpoint
+                const response = await fetch('/chat/text', {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({
+                        message: message,
+                        web_search_enabled: webSearchEnabled,
+                        target_language: selectedLanguage,
+                        system_prompt: systemPrompt
+                    })
+                });
+                const data = await response.json();
+                // Remove typing indicator
+                const indicator = document.getElementById('typing-indicator');
+                if (indicator) indicator.remove();
+                if (data.error) {
+                    showError(data.error);
+                } else {
+                    // Add assistant response
+                    let content = data.response;
+                    if (selectedLanguage && data.language) {
+                        content += ` <span class="language-info">[${data.language}]</span>`;
+                    }
+                    addMessage('assistant', content);
+                }
+            } catch (error) {
+                console.error('Error sending text message:', error);
+                const indicator = document.getElementById('typing-indicator');
+                if (indicator) indicator.remove();
+                showError('메시지 전송 중 오류가 발생했습니다.');
+            }
         }
         function updateStatus(state) {
             if (state === 'connected') {
                 statusText.textContent = '연결됨';
                 sendButton.style.display = 'block';
+                isVoiceActive = true;
             } else if (state === 'connecting') {
                 statusText.textContent = '연결 중...';
                 sendButton.style.display = 'none';
             } else {
                 statusText.textContent = '연결 대기 중';
+                sendButton.style.display = 'block';  // Show send button even when disconnected for text chat
+                isVoiceActive = false;
             }
         }
         function updateButtonState() {
 # Store connection settings
 connection_settings = {}
+# Initialize OpenAI client for text chat
+client = openai.AsyncOpenAI()
 def update_chatbot(chatbot: list[dict], response: ResponseAudioTranscriptDoneEvent):
     chatbot.append({"role": "assistant", "content": response.transcript})
     return chatbot
+def get_translation_instructions(target_language: str) -> str:
+    """Get instructions for translation based on target language"""
+    if not target_language:
+        return ""
+    language_name = SUPPORTED_LANGUAGES.get(target_language, target_language)
+    return (
+        f"\n\nIMPORTANT: You must respond in {language_name} ({target_language}). "
+        f"Translate all your responses to {language_name}."
+    )
+async def process_text_chat(message: str, web_search_enabled: bool, target_language: str,
+                          system_prompt: str) -> Dict[str, str]:
+    """Process text chat using GPT-4o-mini model"""
+    try:
+        # Prepare system message
+        base_instructions = system_prompt or "You are a helpful assistant."
+        translation_instructions = get_translation_instructions(target_language)
+        messages = [
+            {"role": "system", "content": base_instructions + translation_instructions}
+        ]
+        # Handle web search if enabled
+        if web_search_enabled and search_client:
+            # Check if the message requires web search
+            search_keywords = ["날씨", "기온", "비", "눈", "뉴스", "소식", "현재", "최근",
+                             "오늘", "지금", "가격", "환율", "주가", "weather", "news",
+                             "current", "today", "price", "2024", "2025"]
+            should_search = any(keyword in message.lower() for keyword in search_keywords)
+            if should_search:
+                # Perform web search
+                search_results = await search_client.search(message)
+                if search_results:
+                    search_context = "웹 검색 결과:\n\n"
+                    for i, result in enumerate(search_results[:5], 1):
+                        search_context += f"{i}. {result['title']}\n{result['description']}\n\n"
+                    messages.append({
+                        "role": "system",
+                        "content": f"다음 웹 검색 결과를 참고하여 답변하세요:\n\n{search_context}"
+                    })
+        messages.append({"role": "user", "content": message})
+        # Call GPT-4o-mini
+        response = await client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=messages,
+            temperature=0.7,
+            max_tokens=2000
+        )
+        return {
+            "response": response.choices[0].message.content,
+            "language": SUPPORTED_LANGUAGES.get(target_language, "") if target_language else ""
+        }
+    except Exception as e:
+        print(f"Error in text chat: {e}")
+        return {"error": str(e)}
 class OpenAIHandler(AsyncStreamHandler):
     def __init__(self, web_search_enabled: bool = False, target_language: str = "",
                  system_prompt: str = "", webrtc_id: str = None) -> None:
     return response
+@app.post("/chat/text")
+async def chat_text(request: Request):
+    """Handle text chat messages using GPT-4o-mini"""
+    try:
+        body = await request.json()
+        message = body.get("message", "")
+        web_search_enabled = body.get("web_search_enabled", False)
+        target_language = body.get("target_language", "")
+        system_prompt = body.get("system_prompt", "")
+        if not message:
+            return {"error": "메시지가 비어있습니다."}
+        # Process text chat
+        result = await process_text_chat(message, web_search_enabled, target_language, system_prompt)
+        return result
+    except Exception as e:
+        print(f"Error in chat_text endpoint: {e}")
+        return {"error": "채팅 처리 중 오류가 발생했습니다."}
 @app.post("/text_message/{webrtc_id}")
 async def receive_text_message(webrtc_id: str, request: Request):
     """Receive text message from client"""