Update src/streamlit_app.py

src/streamlit_app.py

@@ -10,7 +10,6 @@ import logging
 
 # Backend API Key Configuration
 GEMINI_API_KEY = os.getenv("GEMENI_KEY")
-print(GEMINI_API_KEY)
 
 # Page configuration
 st.set_page_config(
@@ -397,6 +396,16 @@ def main():
                 type="primary",
                 use_container_width=True
             )
+            
+            # Clear results button (only show if results exist)
+            if "analysis_results" in st.session_state and st.session_state["analysis_results"]:
+                if st.button(
+                    "Clear Results",
+                    type="secondary",
+                    use_container_width=True
+                ):
+                    del st.session_state["analysis_results"]
+                    st.rerun()
         
         # Main content area
         if uploaded_video is None:
@@ -444,33 +453,39 @@ def main():
                 )
             
             if json_response:
+                # Store results in session state
+                st.session_state["analysis_results"] = json_response
                 st.success("Analysis complete! Here are your script variations:")
-                
-                # Create tabs for different outputs
-                tab1, tab2, tab3 = st.tabs(["Script Variations", "Video Analysis", "Improvement Recommendations"])
-                
-                with tab1:
-                    display_script_variations(json_response)
-                    
-                    # CSV Download button
-                    csv_content = create_csv_download(json_response)
-                    st.download_button(
-                        label="Download All Scripts (CSV)",
-                        data=csv_content,
-                        file_name="video_script_variations.csv",
-                        mime="text/csv",
-                        type="secondary",
-                        use_container_width=True
-                    )
-                
-                with tab2:
-                    display_video_analysis(json_response)
-                
-                with tab3:
-                    display_timestamp_improvements(json_response)
             else:
                 st.error("Failed to generate script variations. Please try again.")
         
+        # Display results if they exist in session state
+        if "analysis_results" in st.session_state and st.session_state["analysis_results"]:
+            json_response = st.session_state["analysis_results"]
+            
+            # Create tabs for different outputs
+            tab1, tab2, tab3 = st.tabs(["Script Variations", "Video Analysis", "Improvement Recommendations"])
+            
+            with tab1:
+                display_script_variations(json_response)
+                
+                # CSV Download button
+                csv_content = create_csv_download(json_response)
+                st.download_button(
+                    label="Download All Scripts (CSV)",
+                    data=csv_content,
+                    file_name="video_script_variations.csv",
+                    mime="text/csv",
+                    type="secondary",
+                    use_container_width=True
+                )
+            
+            with tab2:
+                display_video_analysis(json_response)
+            
+            with tab3:
+                display_timestamp_improvements(json_response)
+        
         else:
             st.info("Configure your inputs in the sidebar and click 'Generate Script Variations' to begin.")