Update app.py

app.py

@@ -434,7 +434,10 @@ with tab3:
 
         st.divider()
 
-        # Export to CSV
+        # Export section
+        st.subheader("📥 Eksport Wyników")
+        st.markdown("Pobierz wyniki w preferowanym formacie:")
+
         settings = {
             "model": config.get("model"),
             "temperature": config.get("temperature"),
@@ -442,22 +445,103 @@ with tab3:
             "num_responses": config.get("num_responses")
         }
 
-        csv_buffer = test_runner.export_to_csv(
-            st.session_state.rated_responses,
-            results,
-            settings
-        )
-
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"prompt_ab_test_{timestamp}.csv"
 
-        st.download_button(
-            label="💾 Pobierz wyniki (CSV)",
-            data=csv_buffer,
-            file_name=filename,
-            mime="text/csv",
-            use_container_width=True
-        )
+        # Create 3 columns for export buttons
+        col1, col2, col3 = st.columns(3)
+
+        with col1:
+            # CSV Export
+            csv_buffer = test_runner.export_to_csv(
+                st.session_state.rated_responses,
+                results,
+                settings
+            )
+            st.download_button(
+                label="📄 CSV",
+                data=csv_buffer,
+                file_name=f"prompt_ab_test_{timestamp}.csv",
+                mime="text/csv",
+                use_container_width=True,
+                help="Pobierz w formacie CSV"
+            )
+
+            # JSON Export
+            json_buffer = test_runner.export_to_json(
+                st.session_state.rated_responses,
+                results,
+                settings
+            )
+            st.download_button(
+                label="🔧 JSON",
+                data=json_buffer,
+                file_name=f"prompt_ab_test_{timestamp}.json",
+                mime="application/json",
+                use_container_width=True,
+                help="Pobierz w formacie JSON"
+            )
+
+        with col2:
+            # Excel Export
+            excel_buffer = test_runner.export_to_excel(
+                st.session_state.rated_responses,
+                results,
+                settings
+            )
+            st.download_button(
+                label="📊 Excel",
+                data=excel_buffer,
+                file_name=f"prompt_ab_test_{timestamp}.xlsx",
+                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+                use_container_width=True,
+                help="Pobierz w formacie Excel"
+            )
+
+            # TXT Export
+            txt_buffer = test_runner.export_to_txt(
+                st.session_state.rated_responses,
+                results,
+                settings
+            )
+            st.download_button(
+                label="📝 TXT",
+                data=txt_buffer,
+                file_name=f"prompt_ab_test_{timestamp}.txt",
+                mime="text/plain",
+                use_container_width=True,
+                help="Pobierz w formacie TXT"
+            )
+
+        with col3:
+            # Markdown Export
+            markdown_buffer = test_runner.export_to_markdown(
+                st.session_state.rated_responses,
+                results,
+                settings
+            )
+            st.download_button(
+                label="📋 Markdown",
+                data=markdown_buffer,
+                file_name=f"prompt_ab_test_{timestamp}.md",
+                mime="text/markdown",
+                use_container_width=True,
+                help="Pobierz w formacie Markdown"
+            )
+
+            # Word Export
+            word_buffer = test_runner.export_to_word(
+                st.session_state.rated_responses,
+                results,
+                settings
+            )
+            st.download_button(
+                label="📄 Word",
+                data=word_buffer,
+                file_name=f"prompt_ab_test_{timestamp}.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+                use_container_width=True,
+                help="Pobierz w formacie Word"
+            )
 
 # Tab 4: Logs
 with tab4:
@@ -484,7 +568,7 @@ st.divider()
 st.markdown(
     """
     <div style='text-align: center; color: gray; font-size: 0.9em;'>
-    Prompt A/B Tester v2.0 | Created by Heuristica.pl - Marek Staniszewski
+    Prompt A/B Tester v2.0 (Streamlit) | Created by Heuristica.pl - Marek Staniszewski
     </div>
     """,
     unsafe_allow_html=True