Update app.py

app.py

@@ -118,14 +118,6 @@ def generate_similar_question(wrong_q, misconception_id, generator):
                 wrong_answer_text=input_data['wrong_answer_text'],
                 misconception_id=input_data['misconception_id']
             )
-        # Before module 3 integration code
-        # if generated_q:
-        #     return {
-        #         'question': generated_q.question,
-        #         'choices': generated_q.choices,
-        #         'correct': generated_q.correct_answer,
-        #         'explanation': generated_q.explanation
-        #     }
 
         # integrating module3 
         if generated_q:
@@ -164,6 +156,7 @@ def generate_similar_question(wrong_q, misconception_id, generator):
                     # 정답을 세션 스테이트에 저장
                     st.session_state['current_similar_question_answer'] = generated_q.correct_answer
                     return result
+                    
     except Exception as e:
         logger.error(f"Error in generate_similar_question: {str(e)}")
         st.error(f"문제 생성 중 오류가 발생했습니다: {str(e)}")
@@ -364,11 +357,24 @@ def main():
                                 st.session_state[f"selected_answer_{i}"] = None
                                 st.session_state[f"is_correct_{i}"] = None
                                 st.rerun()
+
+                            # 화면 아래 여백 추가
+                            st.markdown("<br>" * 5, unsafe_allow_html=True)  # 5줄의 빈 줄 추가
+                            st.markdown("""
+                                <div style="height: 100px;">
+                                </div>
+                                """, unsafe_allow_html=True)  # 추가 여백
                         else:
                             st.error("유사 문제를 생성할 수 없습니다.")
                             if st.button("❌ 닫기", key=f"close_error_{i}"):
                                 st.session_state[f"show_similar_question_{i}"] = False
                                 st.rerun()
+                                # 화면 아래 여백 추가
+                            st.markdown("<br>" * 5, unsafe_allow_html=True)  # 5줄의 빈 줄 추가
+                            st.markdown("""
+                                <div style="height: 100px;">
+                                </div>
+                                """, unsafe_allow_html=True)  # 추가 여백
 if __name__ == "__main__":
     main()