Update src/streamlit_app.py

src/streamlit_app.py

@@ -151,6 +151,10 @@ def validate_tsv_format(uploaded_file):
     except Exception as e:
         return False, f"Error reading file: {str(e)}"
 
+@st.cache_data
+def convert_df_to_tsv(df):
+    return df.to_csv(sep='\t', index=False).encode('utf-8')
+
 def create_bar_chart(df, view_type):
     """Create interactive bar chart based on view type"""
 
@@ -515,26 +519,37 @@ with st.expander("How to Test Your Model", expanded=True):
     
     if uploaded_file is not None:
         if st.button("🧮 Calculate Score", width="stretch"):
-            with st.spinner("Validating and calculating scores..."):
-                # Validate format
+            # Validate format
+            with st.spinner("Validating upload..."):
                 is_valid, result = validate_tsv_format(uploaded_file)
-                if is_valid:
-                    # Calculate score using hidden ground-truth
-                    scores = calculate_score(result)
-                    if scores is not None:
-                        st.success(f"✅ Score calculated successfully: **{scores['Total Score']:.1f}**")
-                        
-                        # Store in session state
-                        st.session_state.user_results = {
-                            "model": "Your Model",
-                            **scores
-                        }
-                        st.info("👆 Scroll up to see your model on the leaderboard!")
-                        st.rerun()
-                else:
-                    st.error(f"❌ Invalid file format: {result}")
+            if is_valid:
+                # Calculate score using hidden ground-truth
+                scores = calculate_score(result)
+                if scores is not None:
+                    st.success(f"✅ Score calculated successfully: **{scores['score'].mean():.1f}**")
+                    # Store in session state
+                    st.session_state.user_results = {
+                        "model": "Your Model",
+                        **scores
+                    }
+                    st.rerun()
+            else:
+                st.error(f"❌ Invalid file format: {result}")
     else:
         st.info("👆 Upload a TSV file to calculate your score")
+
+    # Allow download of results
+    if 'user_results' in st.session_state
+        user_results = pd.DataFrame([st.session_state.user_results])
+        tsv_data = convert_df_to_tsv(user_results)
+        st.download_button(
+            label="Download scores as TSV",
+            data=tsv_data,
+            file_name='scores.tsv',
+            mime='text/tab-separated-values',
+            width="stretch",
+        )
+        st.info("👆 Scroll up to see your model on the leaderboard!")
     
     # Step 4
     st.markdown("""