Spaces:
Building
Building
| """ | |
| Test script to verify /tmp file upload approach for Hugging Face Spaces | |
| """ | |
| import streamlit as st | |
| import os | |
| import sys | |
| from pathlib import Path | |
| # Add parent directory to path | |
| sys.path.append(os.path.dirname(__file__)) | |
| from web_app.utils import FileUploadHandler | |
| st.title("Test /tmp File Upload Approach") | |
| st.write("This script tests the /tmp directory approach for handling file uploads on Hugging Face Spaces.") | |
| # Show /tmp directory info | |
| st.write("### /tmp Directory Information") | |
| if os.path.exists("/tmp"): | |
| st.write(f"✅ /tmp exists") | |
| st.write(f"- Writable: {os.access('/tmp', os.W_OK)}") | |
| st.write(f"- Readable: {os.access('/tmp', os.R_OK)}") | |
| # Check available space | |
| try: | |
| stat = os.statvfs("/tmp") | |
| free_mb = (stat.f_frsize * stat.f_bavail) / (1024 * 1024) | |
| st.write(f"- Free space: {free_mb:.1f} MB") | |
| except: | |
| st.write("- Free space: Unable to determine") | |
| else: | |
| st.error("❌ /tmp directory does not exist!") | |
| # File upload test | |
| st.write("### File Upload Test") | |
| uploaded_file = st.file_uploader( | |
| "Upload a test file", | |
| type=['txt', 'csv', 'tsv'], | |
| help="Upload any text file to test the /tmp approach" | |
| ) | |
| if uploaded_file: | |
| st.write("**File Details:**") | |
| st.write(f"- Name: {uploaded_file.name}") | |
| st.write(f"- Type: {uploaded_file.type}") | |
| st.write(f"- Size: {uploaded_file.size} bytes") | |
| # Test direct methods | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| st.write("**Direct Methods:**") | |
| # Test read() | |
| try: | |
| uploaded_file.seek(0) | |
| content = uploaded_file.read() | |
| st.write(f"✅ read(): {len(content)} bytes") | |
| except Exception as e: | |
| st.write(f"❌ read(): {str(e)}") | |
| # Test getvalue() | |
| try: | |
| uploaded_file.seek(0) | |
| value = uploaded_file.getvalue() | |
| st.write(f"✅ getvalue(): {len(value)} bytes") | |
| except Exception as e: | |
| st.write(f"❌ getvalue(): {str(e)}") | |
| # Test getbuffer() | |
| try: | |
| buffer = uploaded_file.getbuffer() | |
| st.write(f"✅ getbuffer(): {len(buffer)} bytes") | |
| except Exception as e: | |
| st.write(f"❌ getbuffer(): {str(e)}") | |
| with col2: | |
| st.write("**/tmp Approach:**") | |
| try: | |
| # Save to temp | |
| temp_path = FileUploadHandler.save_to_temp(uploaded_file, prefix="test") | |
| if temp_path: | |
| st.write(f"✅ Saved to: {temp_path}") | |
| # Check file exists | |
| if os.path.exists(temp_path): | |
| file_size = os.path.getsize(temp_path) | |
| st.write(f"✅ File exists: {file_size} bytes") | |
| # Read from temp | |
| content = FileUploadHandler.read_from_temp(temp_path) | |
| if content: | |
| st.write(f"✅ Read content: {len(content)} bytes") | |
| # Show first 100 chars | |
| if isinstance(content, bytes): | |
| preview = content.decode('utf-8', errors='ignore')[:100] | |
| else: | |
| preview = content[:100] | |
| st.text(f"Preview: {preview}...") | |
| else: | |
| st.write("❌ Failed to read content") | |
| # Cleanup | |
| FileUploadHandler.cleanup_temp_file(temp_path) | |
| if not os.path.exists(temp_path): | |
| st.write("✅ Cleanup successful") | |
| else: | |
| st.write("❌ Cleanup failed") | |
| else: | |
| st.write("❌ File not found after saving") | |
| else: | |
| st.write("❌ Failed to save to /tmp") | |
| except Exception as e: | |
| st.error(f"Error: {str(e)}") | |
| import traceback | |
| st.code(traceback.format_exc()) | |
| st.write("---") | |
| st.info("If the /tmp approach works successfully, file uploads should function properly on Hugging Face Spaces.") |