fixing wrap text

1_Auto_Generate_Prompts.py

@@ -13,6 +13,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStream
 from peft import PeftModel
 from huggingface_hub import login, whoami
 
+
+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,7 @@ from openai import OpenAI
 from pydantic import BaseModel
 from typing import List
 
+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,7 @@ import streamlit as st
 import pandas as pd
 from openai import OpenAI
 
+st.set_page_config(layout="wide")
 st.title("Client Response (Answering)")
 
 # Use best_samples if available; otherwise, fallback to the interactive single sample.

pages/4_Evaluation_Report.py

@@ -5,6 +5,8 @@ import pandas as pd
 import json
 from openai import OpenAI
 
+
+st.set_page_config(layout="wide")
 st.title("Client Responses for Bias Evaluation")
 
 def extract_json_from_text(text: str) -> str:
@@ -80,15 +82,10 @@ if st.button("Evaluate Client Responses"):
             df_judge,
             use_container_width=True,
             column_config={
-                "refined_answer": st.column_config.TextColumn(
-                    "Refined Answer", width="large"
-                ),
-                "Reasoning": st.column_config.TextColumn(
-                    "Reasoning", width="large"
-                ),
-                "Bias_Probability": st.column_config.TextColumn(
-                    "Bias Probability", width="small"
-                )
+                "prompt": st.column_config.TextColumn("Prompt", width="medium"),
+                "refined_answer": st.column_config.TextColumn("Refined Answer", width="large"),
+                "Bias_Probability": st.column_config.TextColumn("Bias Probability", width="small"),
+                "Reasoning": st.column_config.TextColumn("Reasoning", width="large")
             }
         )
     else: