Update app.py

app.py

@@ -126,36 +126,35 @@ def fetch_crawl_results(query):
     references = fetch_references(query)
     return references[0], references[1], references[2]
 
-def generate_blog_post(query, prompt_template):
-    try:
-        # 참고글 크롤링
-        references = fetch_references(query)
-        ref1, ref2, ref3 = references
-
-        combined_content = f"참고글1:\n{ref1}\n\n참고글2:\n{ref2}\n\n참고글3:\n{ref3}"
-
-        # 랜덤 시드 생성
-        random_seed = random.randint(1, 2**32 - 1)
-        
-        full_prompt = f"주제: {query}\n\n{prompt_template}\n\n참고 내용:\n{combined_content}"
-
-        response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",
-            messages=[
-                {"role": "system", "content": prompt_template},
-                {"role": "user", "content": full_prompt}
-            ],
-            max_tokens=10000,
-            temperature=0.85,
-            top_p=0.95,
-            frequency_penalty=0.4,
-            presence_penalty=0.4,
-            seed=random_seed
-        )
-        
-        return f"주제: {query}\n\n{response.choices[0].message['content']}", ref1, ref2, ref3
-    except Exception as e:
-        return f"블로그 글 생성 중 오류 발생: {str(e)}", "", "", ""
+def generate_blog_post(query, ref1, ref2, ref3, prompt_template, temperature, top_p, frequency_penalty, presence_penalty):
+    # 크롤링된 참고글 사용
+    combined_content = f"참고글1:\n{ref1}\n\n참고글2:\n{ref2}\n\n참고글3:\n{ref3}"
+    
+    # 전체 프롬프트 구성
+    full_prompt = f"주제: {query}\n\n참고 내용:\n{combined_content}"
+    
+    # OpenAI API 호출
+    response = openai.ChatCompletion.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": prompt_template},
+            {"role": "user", "content": full_prompt}
+        ],
+        max_tokens=10000,
+        temperature=temperature,
+        top_p=top_p,
+        frequency_penalty=frequency_penalty,
+        presence_penalty=presence_penalty,
+        seed=random.randint(1, 2**32 - 1)  # 랜덤 시드 사용
+    )
+    
+    # 생성된 내용 추출
+    generated_content = response.choices[0].message['content']
+    
+    # 주제를 첫 줄에 추가
+    final_content = f"주제: {query}\n\n{generated_content}"
+    
+    return final_content, ref1, ref2, ref3
 
 # PDF 클래스 및 관련 함수 정의
 class PDF(FPDF2):
@@ -258,11 +257,16 @@ with gr.Blocks() as iface:
     pdf_output = gr.File(label="생성된 PDF 파일")
     
     generate_button.click(
-        generate_blog_post,
-        inputs=[query_input, prompt_input],
-        outputs=[output_text, ref1_text, ref2_text, ref3_text],
-        show_progress=True
-    )
+    generate_blog_post,
+    inputs=[
+        query_input, 
+        ref1_input, ref2_input, ref3_input,  
+        prompt_input,
+        temperature_input, top_p_input, frequency_penalty_input, presence_penalty_input
+    ],
+    outputs=[output_text, ref1_text, ref2_text, ref3_text],
+    show_progress=True
+)
     
     save_pdf_button.click(
         save_content_to_pdf,