Upload folder using huggingface_hub

Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy app
+COPY app.py .
+
+# Expose port
+EXPOSE 7860
+
+# Run
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -3,8 +3,8 @@ title: AMK AI Coder Backend
 emoji: 🤖
 colorFrom: purple
 colorTo: indigo
-sdk: streamlit
-sdk_version: 1.40.0
-app_file: app.py
+sdk: docker
+app_file: Dockerfile
 pinned: false
+hardware: t4-small
 ---

app.py

@@ -1,18 +1,21 @@
 """
 AMK AI Coder Platform - Backend API
-Streamlit app for HuggingFace Spaces deployment
+FastAPI for HuggingFace Spaces
 """
 import os
-import asyncio
-from typing import List, Dict
+import json
+from typing import List, Optional, Dict
+from contextlib import asynccontextmanager
 
-import streamlit as st
-import requests
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+import httpx
 
 # Configuration
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "")
 OPENROUTER_API_URL = "https://openrouter.ai/api/v1"
-DEFAULT_MODEL = "inclusionai/ling-2.6-1t"
+DEFAULT_MODEL = "inclusionai/ling-2.6-1t:free"
 
 # System prompts
 DEEP_THINKING_PROMPT = """You are an advanced AI coding assistant with enhanced deep thinking capabilities.
@@ -28,33 +31,82 @@ Provide comprehensive, thoughtful responses."""
 STANDARD_PROMPT = """You are an AI coding assistant. Provide helpful, accurate, and efficient responses.
 Write clean, well-documented code with appropriate comments."""
 
-# Page config
-st.set_page_config(page_title="AMK AI Coder Backend", page_icon="🤖")
+# In-memory conversation storage
+conversations: Dict[str, List[dict]] = {}
 
-# Initialize session state
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-if "deep_thinking" not in st.session_state:
-    st.session_state.deep_thinking = False
-if "model" not in st.session_state:
-    st.session_state.model = DEFAULT_MODEL
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("AMK AI Coder Backend started!")
+    yield
 
-def send_to_openrouter(message: str, model: str, deep_thinking: bool) -> str:
-    """Send message to OpenRouter API"""
+# Create FastAPI app
+app = FastAPI(
+    title="AMK AI Coder Backend API",
+    description="Backend API for AMK AI Coder Platform",
+    version="1.0.0",
+    lifespan=lifespan
+)
+
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Models
+class ChatRequest(BaseModel):
+    message: str
+    session_id: Optional[str] = "default"
+    thinking_mode: bool = False
+    model: str = DEFAULT_MODEL
+
+class ChatResponse(BaseModel):
+    response: str
+    session_id: str
+    thinking_mode: bool
+
+# API Endpoints
+@app.get("/")
+async def root():
+    return {"status": "ok", "message": "AMK AI Coder Backend API", "model": DEFAULT_MODEL}
+
+@app.get("/health")
+async def health():
+    return {
+        "status": "healthy",
+        "api_key_configured": bool(OPENROUTER_API_KEY)
+    }
+
+@app.post("/api/chat", response_model=ChatResponse)
+async def chat(request: ChatRequest):
+    """Main chat endpoint"""
     if not OPENROUTER_API_KEY:
-        return "Error: OpenRouter API key not configured. Please set OPENROUTER_API_KEY in Space settings."
+        raise HTTPException(status_code=500, detail="OpenRouter API key not configured")
+    
+    # Get or create conversation
+    session_id = request.session_id
+    if session_id not in conversations:
+        conversations[session_id] = []
+    
+    messages = conversations[session_id]
     
-    st.session_state.messages.append({"role": "user", "content": message})
+    # Add user message
+    messages.append({"role": "user", "content": request.message})
     
-    system_prompt = DEEP_THINKING_PROMPT if deep_thinking else STANDARD_PROMPT
+    # Build system prompt
+    system_prompt = DEEP_THINKING_PROMPT if request.thinking_mode else STANDARD_PROMPT
     
-    # Build messages with system prompt
-    messages = [{"role": "system", "content": system_prompt}]
-    messages.extend([m for m in st.session_state.messages if m["role"] != "system"][-20:])
+    # Build full context
+    all_messages = [{"role": "system", "content": system_prompt}]
+    all_messages.extend(messages[-20:])  # Keep last 20 messages
     
+    # Call OpenRouter
     payload = {
-        "model": model,
-        "messages": messages,
+        "model": request.model,
+        "messages": all_messages,
         "temperature": 0.7,
         "max_tokens": 4096
     }
@@ -67,66 +119,41 @@ def send_to_openrouter(message: str, model: str, deep_thinking: bool) -> str:
     }
     
     try:
-        response = requests.post(
-            f"{OPENROUTER_API_URL}/chat/completions",
-            headers=headers,
-            json=payload,
-            timeout=120
-        )
-        
-        if response.status_code == 200:
+        async with httpx.AsyncClient(timeout=120.0) as client:
+            response = await client.post(
+                f"{OPENROUTER_API_URL}/chat/completions",
+                headers=headers,
+                json=payload
+            )
+            
+            if response.status_code != 200:
+                raise HTTPException(status_code=response.status_code, detail=response.text)
+            
             data = response.json()
             assistant_message = data["choices"][0]["message"]["content"]
-            st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-            return assistant_message
-        else:
-            return f"API Error: {response.status_code} - {response.text}"
             
+            # Add to conversation history
+            messages.append({"role": "assistant", "content": assistant_message})
+            
+            return ChatResponse(
+                response=assistant_message,
+                session_id=session_id,
+                thinking_mode=request.thinking_mode
+            )
+            
+    except httpx.TimeoutException:
+        raise HTTPException(status_code=504, detail="Request timeout")
     except Exception as e:
-        return f"Error: {str(e)}"
+        raise HTTPException(status_code=500, detail=str(e))
 
-# UI
-st.title("🤖 AMK AI Coder Backend")
-st.markdown("Powered by **OpenRouter** with ling-2.6-1t model")
+@app.post("/api/clear")
+async def clear_chat(session_id: str = "default"):
+    """Clear conversation history"""
+    if session_id in conversations:
+        conversations[session_id] = []
+    return {"status": "cleared", "session_id": session_id}
 
-# Sidebar settings
-with st.sidebar:
-    st.header("Settings")
-    
-    model = st.selectbox(
-        "AI Model",
-        [
-            "inclusionai/ling-2.6-1t:free",
-            "anthropic/claude-3.5-sonnet",
-            "google/gemini-pro",
-        ],
-        index=0
-    )
-    st.session_state.model = model
-    
-    deep_thinking = st.checkbox("🧠 Deep Thinking Mode", value=st.session_state.deep_thinking)
-    st.session_state.deep_thinking = deep_thinking
-    
-    if st.button("🗑️ Clear Chat"):
-        st.session_state.messages = []
-        st.rerun()
-
-# Display chat history
-for message in st.session_state.messages:
-    if message["role"] != "system":
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
-
-# Chat input
-if prompt := st.chat_input("Ask me anything..."):
-    with st.chat_message("user"):
-        st.markdown(prompt)
-    
-    with st.chat_message("assistant"):
-        with st.spinner("Thinking..."):
-            response = send_to_openrouter(prompt, st.session_state.model, st.session_state.deep_thinking)
-            st.markdown(response)
-
-# API Key warning
-if not OPENROUTER_API_KEY:
-    st.warning("⚠️ OpenRouter API key not set. Please add OPENROUTER_API_KEY in Space settings.")
+@app.get("/api/history/{session_id}")
+async def get_history(session_id: str = "default"):
+    """Get conversation history"""
+    return {"session_id": session_id, "messages": conversations.get(session_id, [])}

app_fastapi.py

@@ -0,0 +1,159 @@
+"""
+AMK AI Coder Platform - Backend API
+FastAPI for HuggingFace Spaces
+"""
+import os
+import json
+from typing import List, Optional, Dict
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+import httpx
+
+# Configuration
+OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "")
+OPENROUTER_API_URL = "https://openrouter.ai/api/v1"
+DEFAULT_MODEL = "inclusionai/ling-2.6-1t:free"
+
+# System prompts
+DEEP_THINKING_PROMPT = """You are an advanced AI coding assistant with enhanced deep thinking capabilities.
+When analyzing problems, take time to:
+1. Understand the core requirements
+2. Consider edge cases and potential issues
+3. Plan a structured approach
+4. Write clean, efficient, and well-documented code
+5. Explain your reasoning step by step
+
+Provide comprehensive, thoughtful responses."""
+
+STANDARD_PROMPT = """You are an AI coding assistant. Provide helpful, accurate, and efficient responses.
+Write clean, well-documented code with appropriate comments."""
+
+# In-memory conversation storage
+conversations: Dict[str, List[dict]] = {}
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("AMK AI Coder Backend started!")
+    yield
+
+# Create FastAPI app
+app = FastAPI(
+    title="AMK AI Coder Backend API",
+    description="Backend API for AMK AI Coder Platform",
+    version="1.0.0",
+    lifespan=lifespan
+)
+
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Models
+class ChatRequest(BaseModel):
+    message: str
+    session_id: Optional[str] = "default"
+    thinking_mode: bool = False
+    model: str = DEFAULT_MODEL
+
+class ChatResponse(BaseModel):
+    response: str
+    session_id: str
+    thinking_mode: bool
+
+# API Endpoints
+@app.get("/")
+async def root():
+    return {"status": "ok", "message": "AMK AI Coder Backend API", "model": DEFAULT_MODEL}
+
+@app.get("/health")
+async def health():
+    return {
+        "status": "healthy",
+        "api_key_configured": bool(OPENROUTER_API_KEY)
+    }
+
+@app.post("/api/chat", response_model=ChatResponse)
+async def chat(request: ChatRequest):
+    """Main chat endpoint"""
+    if not OPENROUTER_API_KEY:
+        raise HTTPException(status_code=500, detail="OpenRouter API key not configured")
+    
+    # Get or create conversation
+    session_id = request.session_id
+    if session_id not in conversations:
+        conversations[session_id] = []
+    
+    messages = conversations[session_id]
+    
+    # Add user message
+    messages.append({"role": "user", "content": request.message})
+    
+    # Build system prompt
+    system_prompt = DEEP_THINKING_PROMPT if request.thinking_mode else STANDARD_PROMPT
+    
+    # Build full context
+    all_messages = [{"role": "system", "content": system_prompt}]
+    all_messages.extend(messages[-20:])  # Keep last 20 messages
+    
+    # Call OpenRouter
+    payload = {
+        "model": request.model,
+        "messages": all_messages,
+        "temperature": 0.7,
+        "max_tokens": 4096
+    }
+    
+    headers = {
+        "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+        "Content-Type": "application/json",
+        "HTTP-Referer": "https://amk-coder.vercel.app",
+        "X-Title": "AMK AI Coder Platform"
+    }
+    
+    try:
+        async with httpx.AsyncClient(timeout=120.0) as client:
+            response = await client.post(
+                f"{OPENROUTER_API_URL}/chat/completions",
+                headers=headers,
+                json=payload
+            )
+            
+            if response.status_code != 200:
+                raise HTTPException(status_code=response.status_code, detail=response.text)
+            
+            data = response.json()
+            assistant_message = data["choices"][0]["message"]["content"]
+            
+            # Add to conversation history
+            messages.append({"role": "assistant", "content": assistant_message})
+            
+            return ChatResponse(
+                response=assistant_message,
+                session_id=session_id,
+                thinking_mode=request.thinking_mode
+            )
+            
+    except httpx.TimeoutException:
+        raise HTTPException(status_code=504, detail="Request timeout")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/clear")
+async def clear_chat(session_id: str = "default"):
+    """Clear conversation history"""
+    if session_id in conversations:
+        conversations[session_id] = []
+    return {"status": "cleared", "session_id": session_id}
+
+@app.get("/api/history/{session_id}")
+async def get_history(session_id: str = "default"):
+    """Get conversation history"""
+    return {"session_id": session_id, "messages": conversations.get(session_id, [])}

requirements.txt

@@ -1,2 +1,4 @@
-streamlit>=1.40.0
-requests>=2.31.0
+fastapi>=0.100.0
+uvicorn[standard]>=0.23.0
+httpx>=0.25.0
+pydantic>=2.0.0