Update app.py

app.py

@@ -30,63 +30,67 @@ def main():
     with col2:
         user_input = st.text_input("모르는 문장이나 단어를 입력하세요:", "")
 
-        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:
-            annotated_text(
-                "This ",
-                ("is", ""),
-                " some ",
-                ("annotated", ""),
-                ("text", ""),
-                " for those of ",
-                ("you", ""),
-                " who ",
-                ("like", ""),
-                " this sort of ",
-                ("thing", ""),
-                ". "
-            )"""
+        cols = st.columns(2)
+        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:
+                annotated_text(
+                    "This ",
+                    ("is", ""),
+                    " some ",
+                    ("annotated", ""),
+                    ("text", ""),
+                    " for those of ",
+                    ("you", ""),
+                    " who ",
+                    ("like", ""),
+                    " this sort of ",
+                    ("thing", ""),
+                    ". "
+                )"""
 
-            user_prompt = f"First, extract key words for the topic st-annotated-text format.: {user_text}"
-            
-            messages = [{"role": "system", "content": task_description}, {"role": "user", "content": user_prompt}]
-            
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo-16k",
-                messages=messages,
-                temperature=0.1,
-                max_tokens=2500,
-                top_p=0.2,
-                frequency_penalty=0,
-                presence_penalty=0
-            )
-            
-            highlighted_text = response['choices'][0]['message']['content']
-            
-            exec(highlighted_text)
-        
-        # 추가 설명 버튼을 누르면 추가적인 설명을 해줌
-        if st.button("추가 설명"):
-            # 입력된 텍스트 설명
-            explanation_task = f"Explain the term '{user_input}' in a simple manner, based on the context of the following passage: {user_text}"
-            
-            messages = [
-                {"role": "system", "content": "You are a helpful assistant that explains complex topics in a way that an elementary school student can understand. 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=150
-            )
-            
-            explanation = response['choices'][0]['message']['content']
+                user_prompt = f"First, extract key words for the topic st-annotated-text format.: {user_text}"
+                
+                messages = [{"role": "system", "content": task_description}, {"role": "user", "content": user_prompt}]
+                
+                response = openai.ChatCompletion.create(
+                    model="gpt-3.5-turbo-16k",
+                    messages=messages,
+                    temperature=0.1,
+                    max_tokens=2500,
+                    top_p=0.2,
+                    frequency_penalty=0,
+                    presence_penalty=0
+                )
+                
+                highlighted_text = response['choices'][0]['message']['content']
+                
+                exec(highlighted_text)
+                pass
+
+        with cols[1]:
+            if st.button("추가 설명"):
+                # 입력된 텍스트 설명
+                explanation_task = f"Explain the term '{user_input}' in a simple manner, based on the context of the following passage: {user_text}"
+                
+                messages = [
+                    {"role": "system", "content": "You are a helpful assistant that explains complex topics in a way that an elementary school student can understand. 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=150
+                )
+                
+                explanation = response['choices'][0]['message']['content']
 
-            # 쉬운 표현으로 결과 출력
-            st.text("설명:")
-            st.write(explanation)
-      
+                # 쉬운 표현으로 결과 출력
+                st.text("설명:")
+                st.write(explanation)
+                pass
+        
 if __name__ == "__main__":
     main()