Update app.py

app.py

@@ -32,28 +32,23 @@ def generate_annotated_text(text):
     return response['choices'][0]['message']['content']
 
 #유사한 문장 생성 함수 
-def generate_similar_sentences(base_sentence, num_sentences):
-    generated_sentences = []
-    for _ in range(num_sentences):
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo-16k",
-            messages=[
-                {
-                    "role": "system",
-                    "content": "유사한 문장 생성\n 선택된 문장에 기반하여 학생의 특성 및 활동에 대한 평가 예시 문장을 30개 생성합니다."
-                },
-                {
-                    "role": "user",
-                    "content": base_sentence
-                }
-            ],
-            temperature=1,
-            max_tokens=10000,
-            top_p=1,
-            frequency_penalty=0,
-            presence_penalty=0
-        )
-        generated_sentences.append(response.choices[0].message['content'].strip())
+def generate_similar_sentences(base_sentence):
+    response = openai.ChatCompletion.create(
+    model="gpt-3.5-turbo-16k",
+    messages=[
+            {
+                "role": "system",
+                "content": f"유사한 문장 생성\n {base_sentence}에 기반하여 학생의 특성 및 활동에 대한 평가 예시 문장을 30개 생성합니다."
+            }
+
+        ],
+        temperature=1,
+        max_tokens=10000,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0
+    )
+    generated_sentences.append(response.choices[0].message['content'].strip())
     return generated_sentences
     
     
@@ -100,7 +95,7 @@ if st.button("평가 생성"):
     # 유사한 문장 생성 버튼 추가
     if st.button("유사한 문구 생성"):
         with st.spinner('문장 생성중...'):
-            similar_sentences = generate_similar_sentences(selected_sentence, 30)
+            similar_sentences = generate_similar_sentences(selected_sentence)
             for sentence in similar_sentences:
                 st.write(sentence)