fix streamlit layout

1_Auto_Generate_Prompts.py

@@ -13,6 +13,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStream
 from peft import PeftModel
 from huggingface_hub import login, whoami
 
+
+
+st.set_page_config(layout="wide")
 scroll_css = """
 <style>
 .table-scroll {
@@ -24,7 +27,6 @@ scroll_css = """
 """
 st.markdown(scroll_css, unsafe_allow_html=True)
 
-st.set_page_config(layout="wide")
 st.title("Auto Red Teaming Demo for HI")
 st.markdown(
     """

pages/2_Select_Best_Prompts.py

@@ -7,6 +7,9 @@ from openai import OpenAI
 from pydantic import BaseModel
 from typing import List
 
+
+
+st.set_page_config(layout="wide")
 scroll_css = """
 <style>
 .table-scroll {
@@ -18,7 +21,6 @@ scroll_css = """
 """
 st.markdown(scroll_css, unsafe_allow_html=True)
 
-st.set_page_config(layout="wide")
 st.title("Select Best Prompts")
 
 def extract_json_content(markdown_str: str) -> str:

pages/3_Client_Response.py

@@ -5,6 +5,8 @@ import streamlit as st
 import pandas as pd
 from openai import OpenAI
 
+
+st.set_page_config(layout="wide")
 scroll_css = """
 <style>
 .table-scroll {
@@ -16,7 +18,7 @@ scroll_css = """
 """
 
 st.markdown(scroll_css, unsafe_allow_html=True)
-st.set_page_config(layout="wide")
+
 st.title("Client Response (Answering)")
 
 # Use best_samples if available; otherwise, fallback to the interactive single sample.
@@ -74,7 +76,7 @@ if st.button("Generate responses with Client Application"):
         st.markdown("<div class='table-scroll'>", unsafe_allow_html=True)
         st.table(df_answered_styled)
         st.markdown("</div>", unsafe_allow_html=True)
-        
+
         st.session_state.refined_samples = answered_samples
     else:
         st.error("Please provide your Client API Key.")