Update app.py

app.py

@@ -8,13 +8,22 @@ from src.fine_tune_helpers import fine_tune_model
 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**.
+    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")
@@ -52,7 +61,50 @@ elif app_mode == "URL Title Fetcher":
 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:
-        fine_tune_model(uploaded_file)
+        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