Code Editor for app.py

import streamlit as st
import plotly.express as px
# from issue_recommender import GithubIssueRecommender

st.set_page_config(page_title="GitHub Issue Recommender", layout="wide")

# Initialize the recommender system
@st.cache_resource
def load_recommender():
    return GithubIssueRecommender()

recommender = load_recommender()

# App title and description
st.title("GitHub Issue Recommender")
st.markdown("""
This app helps you find similar GitHub issues using a pre-trained neural network.
Enter a keyword to search for issues, then select a specific issue to get recommendations.
""")

# Search input
search_query = st.text_input("Search GitHub issues by keyword", "").lower()

if search_query:
    # Get search results
    search_results = recommender.search_issues(search_query)
    
    if search_results:
        # Issue selection dropdown
        issue_options = {f"{issue['title']} (#{issue['number']})": idx 
                        for idx, issue in enumerate(search_results)}
        selected_issue = st.selectbox(
            "Select an issue to find similar ones",
            options=list(issue_options.keys())
        )
        
        # Number of recommendations slider
        num_recommendations = st.slider(
            "Number of recommendations",
            min_value=1,
            max_value=10,
            value=5
        )
        
        if selected_issue:
            # Get recommendations
            issue_idx = issue_options[selected_issue]
            recommendations = recommender.get_recommendations(
                search_results[issue_idx],
                n_recommendations=num_recommendations
            )
            
            # Display recommendations
            st.subheader("Similar Issues")
            
            # Create DataFrame for the table
            recommendations_df = pd.DataFrame(
                recommendations,
                columns=["Similarity Score", "Issue Title", "Issue Number", "URL"]
            )
            
            # Display table
            st.dataframe(
                recommendations_df[["Issue Title", "Similarity Score", "URL"]],
                hide_index=False,
                use_container_width=True
            )
            
            # Create bar chart
            fig = px.bar(
                recommendations_df,
                x="Issue Title",
                y="Similarity Score",
                title="Similarity Scores for Recommended Issues",
                labels={"Issue Title": "Issue", "Similarity Score": "Similarity"},
            )
            
            fig.update_layout(
                xaxis_tickangle=-45,
                showlegend=False,
                height=400
            )
            
            st.plotly_chart(fig, use_container_width=True)
    else:
        st.info("No issues found matching your search query. Try different keywords.")