1
import os
2
from fastapi import FastAPI, Depends
3
from fastapi.responses import StreamingResponse
4
from fastapi.middleware.cors import CORSMiddleware
5
from fastapi_clerk_auth import ClerkConfig, ClerkHTTPBearer, HTTPAuthorizationCredentials
6
from openai import OpenAI
7

8
print("[DEBUG] Starting api/index.py import...")
9

10
app = FastAPI()
11

12
# Add CORS middleware
13
app.add_middleware(
14
    CORSMiddleware,
15
    allow_origins=["*"],
16
    allow_credentials=True,
17
    allow_methods=["*"],
18
    allow_headers=["*"],
19
)
20

21
print("[DEBUG] FastAPI app and CORS configured")
22

23
# Configure Clerk authentication
24
clerk_config = ClerkConfig(jwks_url=os.getenv("CLERK_JWKS_URL"))
25
clerk_guard = ClerkHTTPBearer(clerk_config)
26

27
print(f"[DEBUG] Clerk configured with JWKS URL: {os.getenv('CLERK_JWKS_URL')}")
28
print(f"[DEBUG] OpenAI API key present: {bool(os.getenv('OPENAI_API_KEY'))}")
29

30

31
@app.get("/api/health")
32
@app.get("/health")
33
async def health_check():
34
    """Health check endpoint that doesn't require authentication"""
35
    return {"status": "healthy", "service": "api"}
36

37

38
@app.get("/api/idea")
39
@app.get("/idea")
40
def generate_idea(creds: HTTPAuthorizationCredentials = Depends(clerk_guard)):
41
    """
42
    Generate a business idea using OpenAI GPT-4o-mini with streaming response.
43
    Requires Clerk authentication via Authorization header.
44
    """
45
    user_id = creds.decoded["sub"]  # User ID from JWT
46
    print(f"[INFO] Generating idea for user: {user_id}")
47

48
    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
49
    prompt = [
50
        {
51
            "role": "user",
52
            "content": "Reply with a new business idea for AI Agents, formatted with headings, sub-headings and bullet points"
53
        }
54
    ]
55

56
    stream = client.chat.completions.create(
57
        model="gpt-4o-mini",
58
        messages=prompt,
59
        max_tokens=500,
60
        temperature=0.7,
61
        stream=True
62
    )
63

64
    def event_stream():
65
        try:
66
            for chunk in stream:
67
                if chunk.choices[0].delta.content is not None:
68
                    text = chunk.choices[0].delta.content
69
                    # Escape newlines for SSE format
70
                    safe_text = text.replace('
71
', '\
72
')
73
                    yield f"data: {safe_text}
74

75
"
76
            yield "data: [DONE]
77

78
"
79
        except Exception as e:
80
            print(f"[ERROR] Stream error: {e}")
81
            yield f"data: [ERROR] {str(e)}
82

83
"
84

85
    return StreamingResponse(
86
        event_stream(),
87
        media_type="text/event-stream",
88
        headers={
89
            "Cache-Control": "no-cache",
90
            "X-Accel-Buffering": "no"
91
        }
92
    )
93

94

95
@app.get("/api/user")
96
@app.get("/user")
97
def get_user_info(creds: HTTPAuthorizationCredentials = Depends(clerk_guard)):
98
    """Get information about the authenticated user."""
99
    return {
100
        "user_id": creds.decoded.get("sub"),
101
        "email": creds.decoded.get("email"),
102
        "claims": creds.decoded
103
    }
104

105

106
# Vercel serverless function handler
107
handler = app
108

109
print("[DEBUG] Module initialization complete")