Upload folder using huggingface_hub

app.py

@@ -25,7 +25,7 @@ def main():
     # Initialize session state
     if 'optimizer' not in st.session_state:
         st.session_state.optimizer = PromptOptimizer(
-            model=FalImageGenerator(),
+            image_generator=FalImageGenerator(),
             evaluator=LlamaEvaluator(),
             refiner=LlamaPromptRefiner(),
             similarity_metric=LPIPSImageSimilarityMetric(),
@@ -107,9 +107,24 @@ def main():
                         st.text(f"Similarity: {hist_entry['similarity']:.2%}")
                         st.text("Prompt:")
                         st.text(hist_entry['prompt'])
-                        st.text("\nAnalysis:")
-                        for key, value in hist_entry['analysis'].items():
-                            st.text(f"{key}: {value}")
+                        # Toggle analysis view per history entry
+                        expand_key = f"expand_analysis_{idx}"
+                        if 'analysis_expanded' not in st.session_state:
+                            st.session_state['analysis_expanded'] = {}
+                        if expand_key not in st.session_state['analysis_expanded']:
+                            st.session_state['analysis_expanded'][expand_key] = False
+
+                        if st.session_state['analysis_expanded'][expand_key]:
+                            if st.button("Hide Analysis", key=f"hide_{expand_key}"):
+                                st.session_state['analysis_expanded'][expand_key] = False
+                                st.rerun()
+                            st.text("Analysis:")
+                            for key, value in hist_entry['analysis'].items():
+                                st.text(f"{key}: {value}")
+                        else:
+                            if st.button("Expand Analysis", key=expand_key):
+                                st.session_state['analysis_expanded'][expand_key] = True
+                                st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()