Skip to content

Commit 026776b

Browse files
TMHSDigitalTMHSDigital
committed
Add FastAPI web interface and update dependencies
- Introduced FastAPI app with chat and completions endpoints in `web.py`. - Added HTML template for chat interface in `chat.html`. - Included CSS for styling in `style.css`. - Created test script `test_run.py` for API interaction. - Updated `requirements.txt` to include FastAPI and Uvicorn dependencies.
1 parent fd9ee54 commit 026776b

5 files changed

Lines changed: 130 additions & 1 deletion

File tree

‎requirements.txt‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,6 @@ httpx>=0.27.0
22
python-dotenv>=1.0.1
33
pytest>=8.0.0
44
pytest-cov>=5.0.0
5-
pre-commit>=3.0.0
5+
pre-commit>=3.0.0
6+
fastapi>=0.110.0
7+
uvicorn[standard]>=0.29.0

‎src/deepseek_wrapper/static/style.css‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
body { font-family: Arial, sans-serif; background: #f7f7f7; margin: 0; padding: 2em; }
2+
h1 { color: #2b3a67; }
3+
form { margin-bottom: 0.5em; }
4+
input[type=text] { padding: 0.5em; border-radius: 4px; border: 1px solid #ccc; }
5+
button { padding: 0.5em 1em; border-radius: 4px; border: none; background: #2b3a67; color: #fff; cursor: pointer; }
6+
div { margin-bottom: 0.5em; }
7+
.chat-history { background: #fff; border-radius: 8px; padding: 1em; box-shadow: 0 2px 8px #0001; min-height: 120px; }
8+
.user-msg { background: #e3eaff; border-radius: 6px; margin: 0.5em 0; padding: 0.5em 1em; text-align: right; }
9+
.assistant-msg { background: #f0f0f0; border-radius: 6px; margin: 0.5em 0; padding: 0.5em 1em; text-align: left; }

‎src/deepseek_wrapper/templates/chat.html‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
<!DOCTYPE html>
2+
<html lang="en">
3+
<head>
4+
<meta charset="UTF-8">
5+
<title>DeepSeek Chat</title>
6+
<link rel="stylesheet" href="/static/style.css">
7+
</head>
8+
<body>
9+
<h1>DeepSeek Chat</h1>
10+
<form method="post" action="/chat">
11+
<input type="text" name="user_message" placeholder="Type your message..." required style="width: 60%;">
12+
<button type="submit">Send</button>
13+
</form>
14+
<form method="post" action="/completions" style="margin-top: 1em;">
15+
<input type="text" name="prompt" placeholder="Prompt for completions endpoint..." style="width: 60%;">
16+
<button type="submit">Completions</button>
17+
</form>
18+
<form method="post" action="/reset" style="margin-top: 1em;">
19+
<button type="submit">Reset Conversation</button>
20+
</form>
21+
<div class="chat-history" style="margin-top:2em; max-width: 800px;">
22+
{% for msg in messages %}
23+
<div class="{{ msg.role }}-msg"><b>{{ msg.role.title() }}:</b> {{ msg.content }}</div>
24+
{% endfor %}
25+
</div>
26+
</body>
27+
</html>

‎src/deepseek_wrapper/web.py‎

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
from fastapi import FastAPI, Request, Form, Response
2+
from fastapi.responses import HTMLResponse, RedirectResponse
3+
from fastapi.templating import Jinja2Templates
4+
from fastapi.staticfiles import StaticFiles
5+
from starlette.middleware.sessions import SessionMiddleware
6+
from deepseek_wrapper import DeepSeekClient, DeepSeekConfig
7+
import asyncio
8+
import os
9+
10+
app = FastAPI()
11+
app.add_middleware(SessionMiddleware, secret_key=os.getenv("SESSION_SECRET_KEY", "supersecret"))
12+
templates = Jinja2Templates(directory="src/deepseek_wrapper/templates")
13+
app.mount("/static", StaticFiles(directory="src/deepseek_wrapper/static"), name="static")
14+
15+
client = DeepSeekClient()
16+
17+
@app.get("/", response_class=HTMLResponse)
18+
def home(request: Request):
19+
history = request.session.get("history", [])
20+
return templates.TemplateResponse("chat.html", {"request": request, "messages": history})
21+
22+
@app.post("/chat", response_class=HTMLResponse)
23+
def chat(request: Request, user_message: str = Form(...)):
24+
history = request.session.get("history", [])
25+
history.append({"role": "user", "content": user_message})
26+
try:
27+
response = client.chat_completion(history)
28+
except Exception as e:
29+
response = f"Error: {e}"
30+
history.append({"role": "assistant", "content": response})
31+
request.session["history"] = history
32+
return templates.TemplateResponse("chat.html", {"request": request, "messages": history})
33+
34+
@app.post("/completions", response_class=HTMLResponse)
35+
def completions(request: Request, prompt: str = Form(...)):
36+
history = request.session.get("history", [])
37+
try:
38+
response = client.generate_text(prompt)
39+
except Exception as e:
40+
response = f"Error: {e}"
41+
history.append({"role": "user", "content": prompt})
42+
history.append({"role": "assistant", "content": response})
43+
request.session["history"] = history
44+
return templates.TemplateResponse("chat.html", {"request": request, "messages": history})
45+
46+
@app.post("/reset", response_class=HTMLResponse)
47+
def reset(request: Request):
48+
request.session["history"] = []
49+
return RedirectResponse("/", status_code=303)

‎test_run.py‎

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
from deepseek_wrapper import DeepSeekClient, DeepSeekConfig
2+
import asyncio
3+
4+
# 1. Completions endpoint (requires beta base_url)
5+
completions_client = DeepSeekClient(config=DeepSeekConfig(base_url="https://api.deepseek.com/beta"))
6+
print("Sync result (completions, beta API):")
7+
try:
8+
result = completions_client.generate_text("Hello, DeepSeek!", max_tokens=32)
9+
print(result)
10+
except Exception as e:
11+
print(f"Sync error: {e}")
12+
13+
print("\nAsync result (completions, beta API):")
14+
async def main():
15+
try:
16+
result = await completions_client.async_generate_text("Hello, DeepSeek async!", max_tokens=32)
17+
print(result)
18+
except Exception as e:
19+
print(f"Async error: {e}")
20+
asyncio.run(main())
21+
22+
# 2. Chat completion endpoint (works on main API)
23+
chat_client = DeepSeekClient()
24+
print("\nSync result (chat completion, main API):")
25+
try:
26+
chat_result = chat_client.chat_completion([
27+
{"role": "user", "content": "Hello, DeepSeek!"}
28+
])
29+
print(chat_result)
30+
except Exception as e:
31+
print(f"Chat sync error: {e}")
32+
33+
print("\nAsync result (chat completion, main API):")
34+
async def chat_main():
35+
try:
36+
chat_result = await chat_client.async_chat_completion([
37+
{"role": "user", "content": "Hello, DeepSeek async!"}
38+
])
39+
print(chat_result)
40+
except Exception as e:
41+
print(f"Chat async error: {e}")
42+
asyncio.run(chat_main())

0 commit comments

Comments
 (0)