Update app.py

app.py

@@ -17,11 +17,24 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Add custom CSS for animations and styling
+# Add custom CSS with darker text colors and better contrast
 st.markdown("""
     <style>
+        /* Text colors */
+        .stTabs [data-baseweb="tab"] {
+            color: #1a1a1a !important;
+            font-weight: 500;
+        }
+        
+        .stMarkdown, .stText {
+            color: #1a1a1a !important;
+        }
+        
         /* Main container styles */
-        .main-container { padding: 20px; }
+        .main-container { 
+            padding: 20px;
+            color: #1a1a1a;
+        }
         
         /* Agent flow animation */
         .agent-flow {
@@ -33,7 +46,7 @@ st.markdown("""
         }
         
         .agent-node {
-            background-color: white;
+            background-color: #ffffff;
             border-radius: 50%;
             width: 80px;
             height: 80px;
@@ -43,6 +56,10 @@ st.markdown("""
             box-shadow: 0 4px 6px rgba(0,0,0,0.1);
             position: relative;
             z-index: 2;
+            color: #1a1a1a;
+            font-weight: 500;
+            text-align: center;
+            line-height: 1.2;
         }
         
         .agent-connector {
@@ -69,34 +86,17 @@ st.markdown("""
             100% { transform: translateX(100%); opacity: 0; }
         }
         
-        /* Progress animation */
-        @keyframes progress {
-            0% { width: 0; }
-            100% { width: 100%; }
-        }
-        
-        .progress-bar {
-            height: 20px;
-            background-color: #f0f0f0;
-            border-radius: 10px;
-            overflow: hidden;
-            margin: 10px 0;
-        }
-        
-        .progress-fill {
-            height: 100%;
-            background-color: #4CAF50;
-            animation: progress 2s ease-in-out;
-        }
-        
         /* Agent status indicators */
         .agent-status {
-            padding: 10px;
-            margin: 5px 0;
-            border-radius: 5px;
-            background: #f8f9fa;
+            padding: 15px;
+            margin: 10px 0;
+            border-radius: 8px;
+            background: #ffffff;
             border-left: 4px solid #4CAF50;
             animation: slideIn 0.3s ease-out;
+            color: #1a1a1a;
+            font-weight: 400;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.05);
         }
         
         @keyframes slideIn {
@@ -104,34 +104,73 @@ st.markdown("""
             to { transform: translateX(0); opacity: 1; }
         }
         
+        /* Report sections */
+        .report-section {
+            background: #ffffff;
+            padding: 20px;
+            border-radius: 10px;
+            margin: 15px 0;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+            color: #1a1a1a;
+            font-weight: 400;
+            line-height: 1.6;
+        }
+
+        /* Progress bar */
+        .stProgress > div > div {
+            background-color: #4CAF50;
+        }
+        
+        /* Button styles */
+        .stButton > button {
+            font-weight: 500;
+            color: #1a1a1a;
+        }
+        
         /* Tab styling */
         .stTabs [data-baseweb="tab-list"] {
             gap: 24px;
+            background-color: #f8f9fa;
+            padding: 10px;
+            border-radius: 10px;
         }
         
         .stTabs [data-baseweb="tab"] {
             height: 50px;
             white-space: pre-wrap;
-            background-color: #fff;
-            border-radius: 4px;
-            color: #000;
+            background-color: #ffffff;
+            border-radius: 8px;
             padding: 10px 20px;
+            font-weight: 500;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.05);
         }
         
         .stTabs [aria-selected="true"] {
-            background-color: #e6f3ff;
+            background-color: #e7f3ff;
+            color: #0066cc !important;
+        }
+
+        /* Headers */
+        h1, h2, h3, h4, h5, h6 {
+            color: #1a1a1a !important;
+            font-weight: 600;
+        }
+
+        /* Links */
+        a {
+            color: #0066cc !important;
+            text-decoration: none;
         }
         
-        /* Report container */
-        .report-section {
-            background: white;
-            padding: 20px;
-            border-radius: 10px;
-            margin: 10px 0;
-            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        /* Error messages */
+        .stAlert {
+            background-color: #ffebee;
+            color: #c62828;
+            padding: 10px;
+            border-radius: 8px;
         }
     </style>
-""", unsafe_allow_html=True)
+        """, unsafe_allow_html=True)
 
 def create_agent_flow(container):
     """Create animated agent flow diagram"""
@@ -266,45 +305,47 @@ def run_market_research(topic: str, progress_bar, chat_container):
 def main():
     st.title("🤖 AI Market Research Generator")
     
+    if 'current_tab' not in st.session_state:
+        st.session_state.current_tab = "generate"
+    
     # Create tabs
     tab1, tab2 = st.tabs(["Generate Report", "View Report"])
     
     with tab1:
-        col1, col2 = st.columns([2, 3])
-        
-        with col1:
-            st.subheader("Enter Research Topic")
-            topic = st.text_input(
-                "What market would you like to research?",
-                placeholder="e.g., Electric Vehicles Market",
-                key="research_topic"
-            )
+        if st.session_state.current_tab == "generate":
+            col1, col2 = st.columns([2, 3])
             
-            if st.button("Generate Report", type="primary"):
-                if not topic:
-                    st.error("Please enter a research topic")
-                    return
-                
-                # Create progress bar and containers
-                progress = st.progress(0)
-                chat_container = st.container()
+            with col1:
+                st.subheader("Enter Research Topic")
+                topic = st.text_input(
+                    "What market would you like to research?",
+                    placeholder="e.g., Electric Vehicles Market",
+                    key="research_topic"
+                )
                 
-                # Show agent flow diagram
-                create_agent_flow(col2)
-                
-                # Run the research
-                report = run_market_research(topic, progress, chat_container)
-                
-                if report:
-                    # Store results in session state
-                    st.session_state.report = report
-                    st.session_state.current_tab = "report"
+                if st.button("Generate Report", type="primary"):
+                    if not topic:
+                        st.error("Please enter a research topic")
+                        return
+                    
+                    # Create progress bar and containers
+                    progress = st.progress(0)
+                    chat_container = st.container()
                     
-                    # Switch to report tab
-                    st.experimental_rerun()
+                    # Show agent flow diagram
+                    create_agent_flow(col2)
+                    
+                    # Run the research
+                    report = run_market_research(topic, progress, chat_container)
+                    
+                    if report:
+                        # Store results in session state
+                        st.session_state.report = report
+                        st.session_state.current_tab = "report"
+                        st.rerun()  # Using st.rerun() instead of experimental_rerun
     
     with tab2:
-        if hasattr(st.session_state, 'report'):
+        if st.session_state.current_tab == "report" and hasattr(st.session_state, 'report'):
             st.subheader("📊 Market Research Report")
             
             # Display report in sections
@@ -317,11 +358,16 @@ def main():
                         </div>
                     """, unsafe_allow_html=True)
             
+            # Add 'Generate New Report' button
+            if st.button("Generate New Report"):
+                st.session_state.current_tab = "generate"
+                st.rerun()
+            
             # Download button
             st.download_button(
                 label="Download Report",
                 data=st.session_state.report,
-                file_name=f"market_research_{topic.lower().replace(' ', '_')}.md",
+                file_name=f"market_research_{st.session_state.get('research_topic', 'report').lower().replace(' ', '_')}.md",
                 mime="text/markdown"
             )