Spaces:
Running
Running
File size: 4,459 Bytes
c954503 e511bc5 f13ce67 e511bc5 2ce6627 e511bc5 2ce6627 e511bc5 2ce6627 e511bc5 f13ce67 6a46dba e511bc5 f13ce67 e511bc5 2ce6627 e511bc5 2ce6627 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 |
import streamlit as st
import requests
from src.github_analysis import analyze_github_repo
from src.url_fetcher import fetch_url_title
from src.fine_tune_helpers import fine_tune_model
# Title and description
st.title("OSINT Tool 🏢")
st.markdown("""
This tool performs **Open Source Intelligence (OSINT)** analysis on GitHub repositories and fetches titles from URLs.
It also allows uploading datasets (CSV format) for fine-tuning models like **DistilBERT**, **Code Summarization**, **Bug Fixing**, and more.
""")
# Sidebar for navigation
st.sidebar.title("Navigation")
app_mode = st.sidebar.radio("Choose the mode", ["GitHub Repository Analysis", "URL Title Fetcher", "Dataset Upload & Fine-Tuning"])
# List of models for fine-tuning
available_models = [
"semeru/code-text-galeras-code-summarization-3k-deduped",
"semeru/code-code-InjectMutants",
"semeru/code-code-BugFixingSmall",
"semeru/code-code-GeneratingAssertsRaw",
"deepseek-ai/DeepSeek-Prover-V1"
]
# GitHub Repository Analysis
if app_mode == "GitHub Repository Analysis":
st.header("GitHub Repository Analysis")
repo_owner = st.text_input("Enter GitHub Repository Owner", "huggingface")
repo_name = st.text_input("Enter GitHub Repository Name", "transformers")
if st.button("Analyze Repository"):
if repo_owner and repo_name:
repo_data = analyze_github_repo(repo_owner, repo_name)
if repo_data:
st.subheader("Repository Details")
for key, value in repo_data.items():
st.write(f"**{key}**: {value}")
else:
st.error("Failed to retrieve repository details.")
else:
st.warning("Please enter both repository owner and name.")
# URL Title Fetcher
elif app_mode == "URL Title Fetcher":
st.header("URL Title Fetcher")
url = st.text_input("Enter URL", "https://www.huggingface.co")
if st.button("Fetch Title"):
if url:
title = fetch_url_title(url)
if title:
st.write(f"**Page Title**: {title}")
else:
st.error("Failed to retrieve the page title.")
else:
st.warning("Please enter a valid URL.")
# Dataset Upload & Fine-Tuning
elif app_mode == "Dataset Upload & Fine-Tuning":
st.header("Dataset Upload & Fine-Tuning")
# Model selection for fine-tuning
model_choice = st.selectbox("Choose Model for Fine-Tuning", available_models)
# Upload a CSV file for fine-tuning
uploaded_file = st.file_uploader("Upload a CSV file for fine-tuning", type="csv")
if uploaded_file is not None:
st.write(f"Preparing fine-tuning for model: **{model_choice}**")
st.write("File successfully uploaded! Now starting fine-tuning process...")
fine_tune_model(uploaded_file, model_choice) # Assuming the fine_tune_model function handles fine-tuning
# Helper Functions for API Interaction
def analyze_github_repo(owner, repo):
"""Analyzes a GitHub repository and returns information about it."""
try:
response = requests.get(f'https://api.github.com/repos/{owner}/{repo}')
response.raise_for_status()
repo_data = response.json()
return {
"Repository Name": repo_data['name'],
"Owner": repo_data['owner']['login'],
"Stars": repo_data['stargazers_count'],
"Forks": repo_data['forks_count'],
"Issues": repo_data['open_issues_count'],
"Language": repo_data['language'],
"Description": repo_data.get('description', 'No description available.')
}
except requests.exceptions.RequestException as e:
st.error(f"Error fetching GitHub repository: {e}")
return None
def fetch_url_title(url):
"""Fetches the title of a webpage."""
try:
response = requests.get(url)
if response.status_code == 200:
# Extract the title from the HTML content
html_content = response.text
start_index = html_content.find("<title>") + len("<title>")
end_index = html_content.find("</title>")
return html_content[start_index:end_index]
else:
st.error(f"Failed to fetch URL: {response.status_code}")
return None
except requests.exceptions.RequestException as e:
st.error(f"Error fetching URL: {e}")
return None
|