Update app.py

app.py

@@ -15,7 +15,7 @@ openai.api_key = os.getenv("OPENAI_API_KEY")  # 실제 코드에서 주석 해
 def main():
     st.title("한국어 학습자를 위한 HCI tools")
     
-    col1, col2 = st.columns([60, 40], gap="large")
+    col1, col2 = st.columns([70, 30])
 
     with col1:
         st.header("지문")
@@ -30,7 +30,7 @@ def main():
     with col2:
         user_input = st.text_input("모르는 문장이나 단어를 입력하세요:", "")
 
-        cols = st.columns(3)
+        cols = st.columns(4)
         with cols[0]:        
             if st.button("키워드 찾기"):
                 task_description ="""You are a useful helper that generates annotated text for Python's st-annotated-text library. Your task is to identify the topic of the passage and highlight the key words needed to convey the meaning. You should be able to identify the main points. Also, please mark keywords based on the different paragraphs and headings provided in the text. The output should be formatted in the following way:
@@ -82,7 +82,7 @@ def main():
                     model="gpt-3.5-turbo-16k",
                     messages=messages,
                     temperature=0.1,
-                    max_tokens=150
+                    max_tokens=200
                 )
                 
                 explanation = response['choices'][0]['message']['content']
@@ -108,7 +108,30 @@ def main():
                     model="gpt-3.5-turbo-16k",
                     messages=messages,
                     temperature=0.1,
-                    max_tokens=150
+                    max_tokens=500
+                )
+                
+                explanation = response['choices'][0]['message']['content']
+
+                # 쉬운 표현으로 결과 출력
+                st.text("설명:")
+                st.write(explanation)
+                pass
+        #버튼을 누르면 입력중인 글을 다듬어서 다시 써줌
+        with cols[3]:
+            if st.button("다시 쓰기"):
+                explanation_task = f"Rewrite the contents of '{user_input}' so that it will pass the writing test."
+                
+                messages = [
+                    {"role": "system", "content": "You are a helpful assistant. use only korean"},
+                    {"role": "user", "content": explanation_task}
+                ]
+                
+                response = openai.ChatCompletion.create(
+                    model="gpt-3.5-turbo-16k",
+                    messages=messages,
+                    temperature=0.1,
+                    max_tokens=1000
                 )
                 
                 explanation = response['choices'][0]['message']['content']
@@ -118,5 +141,6 @@ def main():
                 st.write(explanation)
                 pass
 
+            
 if __name__ == "__main__":
     main()