Download as txt, pdf, or txt
Download as txt, pdf, or txt
You are on page 1of 2

from flask import Flask, request, jsonify

import openai

app = Flask(__name__)

# Set up OpenAI API key


openai.api_key = 'YOUR_OPENAI_API_KEY'

@app.route('/')
def index():
return """
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>AI Content Generator</title>
</head>
<body>
<h1>AI Content Generator</h1>
<textarea id="prompt" rows="5" cols="50" placeholder="Enter your prompt
here..."></textarea><br>
<button onclick="generateContent()">Generate Content</button><br>
<div id="output"></div>

<script>
async function generateContent() {
const prompt = document.getElementById('prompt').value;
const response = await fetch('/generate', {
method: 'POST',
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({ 'prompt': prompt })
});
const data = await response.json();
document.getElementById('output').innerHTML = '<p>' +
data.generated_text + '</p>';
}
</script>
</body>
</html>
"""

@app.route('/generate', methods=['POST'])
def generate_content():
data = request.get_json()
prompt = data['prompt']

# Generate text using OpenAI's GPT-3 model


response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=100
)

return jsonify({'generated_text': response.choices[0].text.strip()})

if __name__ == '__main__':
app.run(debug=True)

You might also like