fix_plotly_issues.py

"""
Troubleshooting script for Plotly installation and compatibility issues
"""

import subprocess
import sys
import importlib
import os

def check_python_version():
    """Check Python version compatibility"""
    print("🐍 Checking Python version...")
    version = sys.version_info
    print(f"Python version: {version.major}.{version.minor}.{version.micro}")
    
    if version.major < 3 or (version.major == 3 and version.minor < 8):
        print("❌ Python 3.8 or higher is required")
        return False
    else:
        print("✅ Python version is compatible")
        return True

def check_plotly_installation():
    """Check if Plotly is properly installed"""
    print("\n📊 Checking Plotly installation...")
    
    try:
        import plotly
        print(f"✅ Plotly version: {plotly.__version__}")
        
        # Test basic Plotly functionality
        import plotly.graph_objects as go
        import plotly.express as px
        
        # Create a simple test figure
        fig = go.Figure()
        fig.add_trace(go.Scatter(x=[1, 2, 3], y=[1, 2, 3]))
        
        print("✅ Plotly basic functionality working")
        return True
        
    except ImportError as e:
        print(f"❌ Plotly import error: {e}")
        return False
    except Exception as e:
        print(f"❌ Plotly functionality error: {e}")
        return False

def check_streamlit_plotly_integration():
    """Check Streamlit-Plotly integration"""
    print("\n🔄 Checking Streamlit-Plotly integration...")
    
    try:
        import streamlit as st
        print(f"✅ Streamlit version: {st.__version__}")
        
        # Check if streamlit can handle plotly
        import plotly.graph_objects as go
        fig = go.Figure()
        fig.add_trace(go.Scatter(x=[1, 2, 3], y=[1, 2, 3]))
        
        print("✅ Streamlit-Plotly integration should work")
        return True
        
    except Exception as e:
        print(f"❌ Streamlit-Plotly integration error: {e}")
        return False

def install_fixes():
    """Install required packages with fixes"""
    print("\n🔧 Installing fixes...")
    
    packages_to_install = [
        "plotly>=5.0.0",
        "streamlit>=1.28.0",
        "pandas>=1.5.0",
        "numpy>=1.21.0"
    ]
    
    for package in packages_to_install:
        try:
            print(f"Installing {package}...")
            subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", package])
            print(f"✅ {package} installed successfully")
        except subprocess.CalledProcessError as e:
            print(f"❌ Failed to install {package}: {e}")
            return False
    
    return True

def create_minimal_test():
    """Create a minimal test to verify everything works"""
    print("\n🧪 Creating minimal test...")
    
    test_code = '''
import streamlit as st
import plotly.graph_objects as go
import plotly.express as px

st.title("Plotly Test")
st.write("If you can see this and the chart below, Plotly is working!")

# Create a simple chart
fig = px.line(x=[1, 2, 3, 4], y=[1, 4, 2, 3], title="Test Chart")
st.plotly_chart(fig)

st.success("✅ Plotly is working correctly!")
'''
    
    with open("plotly_test.py", "w") as f:
        f.write(test_code)
    
    print("✅ Created plotly_test.py")
    print("Run 'streamlit run plotly_test.py' to test Plotly functionality")

def main():
    """Main troubleshooting function"""
    print("🔍 Plotly Troubleshooting Script")
    print("=" * 50)
    
    # Check Python version
    if not check_python_version():
        print("\n❌ Please upgrade Python to version 3.8 or higher")
        return
    
    # Check Plotly installation
    if not check_plotly_installation():
        print("\n🔧 Attempting to fix Plotly installation...")
        if install_fixes():
            print("✅ Installation fixes applied. Please restart your application.")
        else:
            print("❌ Could not fix installation. Please manually install packages.")
        return
    
    # Check Streamlit integration
    if not check_streamlit_plotly_integration():
        print("\n🔧 Attempting to fix Streamlit integration...")
        install_fixes()
        return
    
    # Create test file
    create_minimal_test()
    
    print("\n" + "=" * 50)
    print("✅ All checks passed! Plotly should be working correctly.")
    print("\nIf you're still having issues:")
    print("1. Restart your terminal/command prompt")
    print("2. Run: pip install --upgrade plotly streamlit")
    print("3. Run: streamlit run plotly_test.py")
    print("4. If that works, try: streamlit run app.py")

if __name__ == "__main__":
    main()