import streamlit as st
import plotly.graph_objs as go
from github_analytics.singular_analysis_chat import predict_2vars


def create_line_chart(repo_data):
    # Extract labels and data from repo_data
    if not repo_data:
        return None

    labels = [repo['name'] for repo in repo_data]
    forks = [repo['forks_count'] for repo in repo_data]

    # Create a Plotly line chart
    fig = go.Figure(data=go.Scatter(x=labels, y=forks, mode='lines+markers'))

    # Customize the chart
    fig.update_layout(
        title='Forks by Repository',
        xaxis_title='Repository',
        yaxis_title='Number of Forks',
        hovermode='closest',
        xaxis=dict(tickangle=-45),  # Rotate x-axis labels
        width=800,  # Set desired width
        height=600
    )

    # Customize hover label
    fig.update_traces(
        hovertemplate='Repository: %{x}<br>Forks: %{y}'
    )
    response = predict_2vars(labels, forks, "repo names", "forks")
    return st.plotly_chart(fig), st.write(response)