Update app.py

app.py

@@ -4,12 +4,15 @@ import os
 from dotenv import load_dotenv
 from styles import get_custom_css
 
+# Set page to wide mode to use full width
+st.set_page_config(layout="wide")
+
 # Load environment variables
 load_dotenv()
 
 # Configure Google Gemini API
 genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
-model = genai.GenerativeModel('gemini-pro')
+model = genai.GenerativeModel('gemini-2.0-flash')
 
 # Hide Streamlit menu and footer
 st.markdown("""