-
Notifications
You must be signed in to change notification settings - Fork 9
/
leetcode.py
29 lines (25 loc) · 993 Bytes
/
leetcode.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import streamlit as st
import os
import openai
st.header("Leet Code Solution generator in python")
leetcode_question=st.text_area("Type Problem in following format <type Leet Code with constraints> ")
button= st.button("Fetch Python code")
def gen_auto_response(leetcode_question):
openai.api_key = "Enter your OPEN AI API key"
response = openai.Completion.create(
model="code-cushman-001",
prompt=f""""Given a Python solution for the leetcode question below
Leet Code Question: {leetcode_question}
Python Solution: """,
temperature=0,
max_tokens=1124,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
print(response)
return response.choices[0].text
if leetcode_question and button:
with st.spinner("Generating Python Solution to your leetcode problem Please Wait"):
reply=gen_auto_response(leetcode_question)
st.code(reply)