arxivgpt kim commited on
Commit
7f3a634
·
verified ·
1 Parent(s): ebb52b4

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -0
app.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import openai
3
+
4
+ # OpenAI API 키와 Assistant ID 설정
5
+ OPENAI_API_KEY = "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"
6
+ ASSISTANT_ID = "asst_GLnbQ7HdAoRtQL3IKiODpRFB"
7
+
8
+ # OpenAI API 키 설정
9
+ openai.api_key = OPENAI_API_KEY
10
+
11
+ def ask_openai(prompt):
12
+ response = openai.ChatCompletion.create(
13
+ model="gpt-4-turbo-preview", # 'gpt-4-turbo-preview' 모델 사용
14
+ messages=[
15
+ {"role": "system", "content": "You are a helpful assistant."},
16
+ {"role": "user", "content": prompt},
17
+ ],
18
+ temperature=0.7,
19
+ )
20
+ return response.choices[0].message['content']
21
+
22
+ # Gradio UI 구성
23
+ interface = gr.Interface(
24
+ fn=ask_openai,
25
+ inputs=gr.inputs.Textbox(lines=2, placeholder="여기에 질문을 입력하세요..."),
26
+ outputs="text",
27
+ title="OpenAI GPT-4 Chat",
28
+ description="OpenAI GPT-4를 사용하여 답변을 제공하는 챗봇입니다."
29
+ )
30
+
31
+ # Gradio 앱 실행
32
+ interface.launch()