Update app.py

app.py

@@ -39,62 +39,103 @@ def main():
 
         if btn_keyword:
             # 키워드 찾기 로직
-            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.
+            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",
-                prompt=f"Extract key words from the text: {user_text}",
+                messages=messages,
                 temperature=0.1,
-                max_tokens=200
+                max_tokens=2500,
+                top_p=0.2,
+                frequency_penalty=0,
+                presence_penalty=0
             )
-            keywords = response['choices'][0]['text']
-
-            keywords = "민주주의, 국민, 정치, 참여, 권리, 참정권, 선거, 사회 집단, 대표자, 공직자, 대표성, 투표율, 정당성"  # 예시
-            st.write(f"키워드: {keywords}")
+            
+            highlighted_text = response['choices'][0]['message']['content']
+            
+            exec(highlighted_text)
+            pass
+            
 
         if btn_explanation:
-            # 추가 설명 로직
-            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.
-
+            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",
-                prompt=f"Explain the term '{user_input}' in the context of the text: {user_text}",
+                messages=messages,
                 temperature=0.1,
                 max_tokens=200
             )
-            explanation = response['choices'][0]['text']
+            
+            explanation = response['choices'][0]['message']['content']
 
-            explanation = "여기에 추가 설명이 들어갑니다."  # 예시
-            st.write(f"설명: {explanation}")
+            # 쉬운 표현으로 결과 출력
+            st.write(f"추가 설명: {explanation}")
+            pass
+        
 
         if btn_simple:
-            # 쉬운 표현 로직
-            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.
- 
+            explanation_task = f"Describe the fingerprint of '{user_text}' in a way that an elementary school student could understand."
+                
+            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",
-                prompt=f"Rewrite the text in simpler terms: {user_text}",
+                messages=messages,
                 temperature=0.1,
-                max_tokens=200
+                max_tokens=500
             )
-            simple_text = response['choices'][0]['text']
-            simple_text = "여기에 쉬운 표현이 들어갑니다."  # 예시
-            st.write(f"쉬운 표현: {simple_text}")
+            
+            explanation = response['choices'][0]['message']['content']
 
+            # 쉬운 표현으로 결과 출력
+            st.write(f"쉬운 글: {explanation}")
+            pass
+            
         if btn_rewrite:
-            # 다시 쓰기 로직
-            # OpenAI API 호출은 실제로 실행하실 때 주석을 해제해주세요.
+            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",
-                prompt=f"Rewrite the text: {user_text}",
+                messages=messages,
                 temperature=0.1,
-                max_tokens=200
+                max_tokens=1000
             )
-            rewritten_text = response['choices'][0]['text']
             
-            rewritten_text = "여기에 다시 쓴 텍스트가 들어갑니다."  # 예시
-            st.write(f"다시 쓰기: {rewritten_text}")
+            explanation = response['choices'][0]['message']['content']
+
+            st.write(f"다시 쓰기: {explanation}")
 
 if __name__ == "__main__":
     main()