Spaces:

openfree
/

Chain-of-Recursive-Thoughts

Running

App Files Files Community

openfree commited on 17 days ago

Commit

75e8c43

verified ·

1 Parent(s): baa1dde

Update app-backup.py

Browse files

Files changed (1) hide show

app-backup.py +52 -49

app-backup.py CHANGED Viewed

@@ -12,11 +12,11 @@ from typing import List, Dict, Optional, Any
 import logging
 import uuid
-# 모듈이 존재하지 않으면 현재 디렉토리에서 가져오도록 시도
 try:
     from recursive_thinking_ai import EnhancedRecursiveThinkingChat
 except ModuleNotFoundError:
-    # 현재 디렉토리에 recursive_thinking_ai.py 파일이 있어야 함
     import sys
     sys.path.append('.')
     from recursive_thinking_ai import EnhancedRecursiveThinkingChat
@@ -29,8 +29,8 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 app = FastAPI(
-    title="Chain of Recursive Thoughts",
-    description="API for Enhanced Recursive Thinking Chat",
     version="1.0.0"
 )
@@ -46,14 +46,14 @@ app.add_middleware(
 # Create a dictionary to store chat instances
 chat_instances = {}
-# 환경 변수에서 API 키 가져오기
 API_KEY = os.getenv("OPENROUTE_API")
 if not API_KEY:
-    logger.warning("OPENROUTE_API 환경 변수가 설정되지 않았습니다. 일부 기능이 동작하지 않을 수 있습니다.")
 # Pydantic models for request/response validation
 class ChatConfig(BaseModel):
-    # api_key 항목은 제거하고, 모델과 temperature만 받도록 수정
     model: str = "mistralai/mistral-small-3.1-24b-instruct:free"
     temperature: Optional[float] = Field(default=0.7, ge=0.0, le=1.0)
@@ -82,7 +82,7 @@ class InitResponse(BaseModel):
     session_id: str
     status: str
-# 간단한 HTML 인터페이스 제공 (API 키 입력 폼을 제거함)
 @app.get("/", response_class=HTMLResponse)
 async def root():
     """Root endpoint with a simple HTML interface"""
@@ -90,7 +90,7 @@ async def root():
     <!DOCTYPE html>
     <html>
     <head>
-        <title>RecThink API</title>
         <style>
             body {{
                 font-family: Arial, sans-serif;
@@ -149,48 +149,53 @@ async def root():
         </style>
     </head>
     <body>
-        <h1>RecThink API 인터페이스</h1>
         <div class="container">
             <div id="init-form">
-                <h2>1. 채팅 초기화</h2>
-                <!-- API 키 입력란 제거 -->
-                <label for="model">모델:</label>
                 <input type="text" id="model" value="mistralai/mistral-small-3.1-24b-instruct:free">
-                <label for="temperature">온도 (Temperature):</label>
                 <input type="number" id="temperature" min="0" max="1" step="0.1" value="0.7">
-                <button onclick="initializeChat()">초기화</button>
             </div>
             <div id="chat-form" style="display: none;">
-                <h2>2. 메시지 전송</h2>
-                <p>세션 ID: <span id="session-id"></span></p>
-                <label for="message">메시지:</label>
-                <textarea id="message" rows="4" placeholder="메시지를 입력하세요"></textarea>
-                <label for="thinking-rounds">사고 라운드 (선택사항):</label>
-                <input type="number" id="thinking-rounds" min="1" max="10" placeholder="자동 결정">
-                <label for="alternatives">대안 개수 (선택사항):</label>
                 <input type="number" id="alternatives" min="1" max="5" value="3">
-                <button onclick="sendMessage()">전송</button>
-                <button onclick="resetChat()" style="background-color: #f44336;">초기화</button>
             </div>
             <div id="response-container" style="display: none;">
-                <h2>3. 응답</h2>
-                <div id="response">응답이 여기에 표시됩니다...</div>
                 <div class="log">
-                    <h3>생각 과정 로그:</h3>
                     <div id="thinking-log"></div>
                 </div>
             </div>
         </div>
         <script>
             let currentSessionId = null;
@@ -219,16 +224,16 @@ async def root():
                         document.getElementById('chat-form').style.display = 'block';
                         document.getElementById('response-container').style.display = 'block';
                     }} else {{
-                        alert('초기화 실패: ' + (data.detail || '알 수 없는 오류'));
                     }}
                 }} catch (error) {{
-                    alert('오류 발생: ' + error.message);
                 }}
             }}
             async function sendMessage() {{
                 if (!currentSessionId) {{
-                    alert('먼저 채팅을 초기화해주세요');
                     return;
                 }}
@@ -237,11 +242,11 @@ async def root():
                 const alternatives = document.getElementById('alternatives').value;
                 if (!message) {{
-                    alert('메시지를 입력해주세요');
                     return;
                 }}
-                document.getElementById('response').textContent = '처리 중...';
                 document.getElementById('thinking-log').textContent = '';
                 try {{
@@ -266,22 +271,21 @@ async def root():
                         // Display thinking history
                         let thinkingLogHTML = '';
                         data.thinking_history.forEach(item => {{
-                            const selected = item.selected ? ' ✓ 선택됨' : '';
-                            // 수정된 부분: 문자열 연결 방식으로 전환
-                            thinkingLogHTML += "<p><strong>라운드 " + item.round + selected + ":</strong> ";
                             if (item.explanation && item.selected) {{
-                                thinkingLogHTML += "<br><em>선택 이유: " + item.explanation + "</em>";
                             }}
                             thinkingLogHTML += "</p>";
                         }});
                         document.getElementById('thinking-log').innerHTML = thinkingLogHTML;
                     }} else {{
-                        document.getElementById('response').textContent = '오류: ' + (data.detail || '알 수 없는 오류');
                     }}
                 }} catch (error) {{
-                    document.getElementById('response').textContent = '오류 발생: ' + error.message;
                 }}
             }}
@@ -300,7 +304,7 @@ async def root():
     """
     return html_content
-# Healthcheck endpoint
 @app.get("/health")
 async def health_check():
     """Health check endpoint"""
@@ -308,14 +312,14 @@ async def health_check():
 @app.post("/api/initialize", response_model=InitResponse)
 async def initialize_chat(config: ChatConfig):
-    """Initialize a new chat session using environment API key"""
     try:
         # Generate a session ID
         session_id = f"session_{datetime.now().strftime('%Y%m%d%H%M%S')}_{uuid.uuid4().hex[:8]}"
-        # 환경 변수가 없을 경우 에러 처리(또는 경고)
         if not API_KEY:
-            raise HTTPException(status_code=400, detail="OPENROUTE_API 환경 변수가 설정되지 않았습니다.")
         # Initialize the chat instance
         chat = EnhancedRecursiveThinkingChat(
@@ -336,7 +340,7 @@ async def initialize_chat(config: ChatConfig):
 @app.post("/api/send_message")
 async def send_message(request: MessageRequest):
-    """Send a message and get a response with thinking process"""
     try:
         if request.session_id not in chat_instances:
             raise HTTPException(status_code=404, detail="Session not found")
@@ -387,7 +391,7 @@ async def send_message(request: MessageRequest):
 @app.post("/api/save")
 async def save_conversation(request: SaveRequest):
-    """Save the conversation or full thinking log"""
     try:
         if request.session_id not in chat_instances:
             raise HTTPException(status_code=404, detail="Session not found")
@@ -486,7 +490,7 @@ class ConnectionManager:
 manager = ConnectionManager()
-# WebSocket for streaming thinking process
 @app.websocket("/ws/{session_id}")
 async def websocket_endpoint(websocket: WebSocket, session_id: str):
     try:
@@ -499,7 +503,7 @@ async def websocket_endpoint(websocket: WebSocket, session_id: str):
         chat = chat_instances[session_id]["chat"]
-        # Set up a custom callback to stream thinking process
         original_call_api = chat._call_api
         async def stream_callback(chunk):
@@ -592,16 +596,15 @@ async def websocket_endpoint(websocket: WebSocket, session_id: str):
         except:
             pass
     finally:
-        # Restore original method if needed
         if 'chat' in locals() and 'original_call_api' in locals():
             chat._call_api = original_call_api
         # Make sure to disconnect
         manager.disconnect(session_id)
-# 포트 설정 - 허깅페이스 스페이스에서는 7860 포트를 사용해야 함
 if __name__ == "__main__":
-    # 허깅페이스 스페이스에서 실행시 포트 7860 사용
     port = 7860
     print(f"Starting server on port {port}")
     uvicorn.run("app:app", host="0.0.0.0", port=port)

 import logging
 import uuid
+# If the module does not exist, try to import from the current directory
 try:
     from recursive_thinking_ai import EnhancedRecursiveThinkingChat
 except ModuleNotFoundError:
+    # The file recursive_thinking_ai.py must exist in the current directory
     import sys
     sys.path.append('.')
     from recursive_thinking_ai import EnhancedRecursiveThinkingChat
 logger = logging.getLogger(__name__)
 app = FastAPI(
+    title="Chain-of-Recursive-Thoughts: TEST",
+    description="https://github.com/PhialsBasement/Chain-of-Recursive-Thoughts",
     version="1.0.0"
 )
 # Create a dictionary to store chat instances
 chat_instances = {}
+# Retrieve API key from environment variable
 API_KEY = os.getenv("OPENROUTE_API")
 if not API_KEY:
+    logger.warning("The OPENROUTE_API environment variable is not set. Some features may not work.")
 # Pydantic models for request/response validation
 class ChatConfig(BaseModel):
+    # Removed api_key field; only model and temperature are received
     model: str = "mistralai/mistral-small-3.1-24b-instruct:free"
     temperature: Optional[float] = Field(default=0.7, ge=0.0, le=1.0)
     session_id: str
     status: str
+# Simple HTML interface (API key input form removed)
 @app.get("/", response_class=HTMLResponse)
 async def root():
     """Root endpoint with a simple HTML interface"""
     <!DOCTYPE html>
     <html>
     <head>
+        <title>Chain-of-Recursive-Thoughts: TEST</title>
         <style>
             body {{
                 font-family: Arial, sans-serif;
         </style>
     </head>
     <body>
+        <h1>Chain-of-Recursive-Thoughts: TEST</h1>
         <div class="container">
             <div id="init-form">
+                <h2>1. Initialize Chat</h2>
+                <!-- API key input removed -->
+                <label for="model">Model:</label>
                 <input type="text" id="model" value="mistralai/mistral-small-3.1-24b-instruct:free">
+                <label for="temperature">Temperature:</label>
                 <input type="number" id="temperature" min="0" max="1" step="0.1" value="0.7">
+                <button onclick="initializeChat()">Initialize</button>
             </div>
             <div id="chat-form" style="display: none;">
+                <h2>2. Send Message</h2>
+                <p>Session ID: <span id="session-id"></span></p>
+                <label for="message">Message:</label>
+                <textarea id="message" rows="4" placeholder="Enter your message"></textarea>
+                <label for="thinking-rounds">Thinking Rounds (optional):</label>
+                <input type="number" id="thinking-rounds" min="1" max="10" placeholder="Auto">
+                <label for="alternatives">Number of Alternatives (optional):</label>
                 <input type="number" id="alternatives" min="1" max="5" value="3">
+                <button onclick="sendMessage()">Send</button>
+                <button onclick="resetChat()" style="background-color: #f44336;">Reset</button>
             </div>
             <div id="response-container" style="display: none;">
+                <h2>3. Response</h2>
+                <div id="response">The response will appear here...</div>
                 <div class="log">
+                    <h3>Thinking Process Log:</h3>
                     <div id="thinking-log"></div>
                 </div>
             </div>
         </div>
+        <div style="margin-top: 30px;">
+            <p>Repo: https://github.com/PhialsBasement/Chain-of-Recursive-Thoughts</p>
+            <p>Community: https://discord.gg/openfreeai</p>
+        </div>
         <script>
             let currentSessionId = null;
                         document.getElementById('chat-form').style.display = 'block';
                         document.getElementById('response-container').style.display = 'block';
                     }} else {{
+                        alert('Initialization failed: ' + (data.detail || 'Unknown error'));
                     }}
                 }} catch (error) {{
+                    alert('An error occurred: ' + error.message);
                 }}
             }}
             async function sendMessage() {{
                 if (!currentSessionId) {{
+                    alert('Please initialize a chat session first.');
                     return;
                 }}
                 const alternatives = document.getElementById('alternatives').value;
                 if (!message) {{
+                    alert('Please enter a message.');
                     return;
                 }}
+                document.getElementById('response').textContent = 'Processing...';
                 document.getElementById('thinking-log').textContent = '';
                 try {{
                         // Display thinking history
                         let thinkingLogHTML = '';
                         data.thinking_history.forEach(item => {{
+                            const selected = item.selected ? ' ✓ Selected' : '';
+                            thinkingLogHTML += "<p><strong>Round " + item.round + selected + ":</strong> ";
                             if (item.explanation && item.selected) {{
+                                thinkingLogHTML += "<br><em>Reason for selection: " + item.explanation + "</em>";
                             }}
                             thinkingLogHTML += "</p>";
                         }});
                         document.getElementById('thinking-log').innerHTML = thinkingLogHTML;
                     }} else {{
+                        document.getElementById('response').textContent = 'Error: ' + (data.detail || 'Unknown error');
                     }}
                 }} catch (error) {{
+                    document.getElementById('response').textContent = 'An error occurred: ' + error.message;
                 }}
             }}
     """
     return html_content
+# Health check endpoint
 @app.get("/health")
 async def health_check():
     """Health check endpoint"""
 @app.post("/api/initialize", response_model=InitResponse)
 async def initialize_chat(config: ChatConfig):
+    """Initialize a new chat session using the environment API key"""
     try:
         # Generate a session ID
         session_id = f"session_{datetime.now().strftime('%Y%m%d%H%M%S')}_{uuid.uuid4().hex[:8]}"
+        # If the environment variable is missing, raise an error (or warning)
         if not API_KEY:
+            raise HTTPException(status_code=400, detail="The OPENROUTE_API environment variable is not set.")
         # Initialize the chat instance
         chat = EnhancedRecursiveThinkingChat(
 @app.post("/api/send_message")
 async def send_message(request: MessageRequest):
+    """Send a message and get a response with the thinking process"""
     try:
         if request.session_id not in chat_instances:
             raise HTTPException(status_code=404, detail="Session not found")
 @app.post("/api/save")
 async def save_conversation(request: SaveRequest):
+    """Save the conversation or the full thinking log"""
     try:
         if request.session_id not in chat_instances:
             raise HTTPException(status_code=404, detail="Session not found")
 manager = ConnectionManager()
+# WebSocket for streaming the thinking process
 @app.websocket("/ws/{session_id}")
 async def websocket_endpoint(websocket: WebSocket, session_id: str):
     try:
         chat = chat_instances[session_id]["chat"]
+        # Set up a custom callback to stream the thinking process
         original_call_api = chat._call_api
         async def stream_callback(chunk):
         except:
             pass
     finally:
+        # Restore the original method if needed
         if 'chat' in locals() and 'original_call_api' in locals():
             chat._call_api = original_call_api
         # Make sure to disconnect
         manager.disconnect(session_id)
+# Use port 7860 for Hugging Face Spaces
 if __name__ == "__main__":
     port = 7860
     print(f"Starting server on port {port}")
     uvicorn.run("app:app", host="0.0.0.0", port=port)