from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse, JSONResponse
import openai
import os
app = FastAPI()
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
@app.get("/", response_class=HTMLResponse)
async def root():
return """
SuperDeepSite: AI Website Generator
SuperDeepSite 🚀
"""
@app.post("/generate", response_class=HTMLResponse)
async def generate(request: Request):
form = await request.form()
user_prompt = form.get("prompt", "")
if not OPENAI_API_KEY:
return HTMLResponse(
"Error: Set OPENAI_API_KEY as a secret/environment variable in your Space settings."
)
openai.api_key = OPENAI_API_KEY
prompt = f"Generate Python Flask code for a web app that: {user_prompt}\nGive only the code."
try:
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a code-writing assistant."},
{"role": "user", "content": prompt}
],
max_tokens=700,
)
code_output = completion.choices[0].message.content
code_html = f"{code_output}"
except Exception as e:
code_html = f"Error: {str(e)}"
return HTMLResponse(
f"Result
{code_html}
← Go Back"
)