Update app.py

app.py

@@ -1,37 +1,46 @@
 import streamlit as st
 import requests
-import pandas as pd
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
+from transformers import MarianMTModel, MarianTokenizer
 
 # Set up the Streamlit page
 st.title("AI Opportunity Finder for Youth")
 st.write("Find Scholarships, Internships, Online Courses, and more!")
 
-# Function to get scholarships data from a mock API
+# Language Translation Function
+def translate_text(text, target_lang='de'):
+    # Use Hugging Face's MarianMT for translation
+    model_name = f'Helsinki-NLP/opus-mt-en-{target_lang}'
+    model = MarianMTModel.from_pretrained(model_name)
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+
+    translated = model.generate(**tokenizer(text, return_tensors="pt", padding=True, truncation=True))
+    translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
+    return translated_text
+
+# Mock function to get data from APIs (replace with actual API calls)
 def get_scholarships(country, interests):
-    # Example: Replace with a real API URL (mocked for demonstration)
-    url = f"https://jsonplaceholder.typicode.com/posts"  # Mock API
-    response = requests.get(url)
+    url = f"https://jsonplaceholder.typicode.com/posts"  # Mock API (replace with real one)
     
-    if response.status_code == 200:
-        posts = response.json()[:5]  # Take only the first 5 posts as mock scholarships
-        return [{"title": f"Scholarship {i+1}", "description": post.get('body', 'No description available'), "eligibility": "Any student from any background."} for i, post in enumerate(posts)]
+    # Simulate API response based on country
+    if country == "USA":
+        return [{"title": f"USA Scholarship {i+1}", "description": f"Description for scholarship {i+1} in USA.", "eligibility": "Any student from USA."} for i in range(5)]
+    elif country == "Germany":
+        return [{"title": f"Germany Scholarship {i+1}", "description": f"Description for scholarship {i+1} in Germany.", "eligibility": "Any student from Germany."} for i in range(5)]
     else:
-        return []
+        return [{"title": f"Scholarship {i+1}", "description": f"Description for scholarship {i+1} in {country}.", "eligibility": "Any student from any background."} for i in range(5)]
 
-# Function to get internships data from a mock API
 def get_internships(country):
-    # Example: Replace with a real API URL (mocked for demonstration)
     url = f"https://jsonplaceholder.typicode.com/posts"  # Mock API for testing
-    response = requests.get(url)
-    
-    if response.status_code == 200:
-        return [{"jobtitle": f"Internship {i+1}", "company": "Sample Company", "location": "Remote", "snippet": "Description of the internship."} for i in range(5)]
+    # Simulate internships data
+    if country == "USA":
+        return [{"jobtitle": f"Internship {i+1}", "company": "USA Company", "location": "USA", "snippet": "Description of internship in USA."} for i in range(5)]
+    elif country == "Germany":
+        return [{"jobtitle": f"Internship {i+1}", "company": "Germany Company", "location": "Germany", "snippet": "Description of internship in Germany."} for i in range(5)]
     else:
-        return []
+        return [{"jobtitle": f"Internship {i+1}", "company": "Sample Company", "location": "Remote", "snippet": "Description of internship."} for i in range(5)]
 
-# Function to recommend opportunities based on user input
 def recommend_opportunities(user_interests, user_skills, opportunities):
     user_profile = [f"{user_interests} {user_skills}"]
     opportunities_text = [f"{opportunity.get('description', 'No description available')} {opportunity.get('eligibility', 'No eligibility available')}" for opportunity in opportunities]
@@ -51,9 +60,10 @@ def recommend_opportunities(user_interests, user_skills, opportunities):
 # Form to gather user profile and country selection
 with st.form(key='user_form'):
     st.sidebar.header("User Profile")
-    location = st.selectbox("Select your Country", ["Pakistan", "USA", "Germany", "India", "UK", "Australia"])  # Add more countries as needed
+    location = st.selectbox("Select your Country", ["USA", "Germany", "UK", "India", "Australia", "Pakistan"])  # You can add more countries here
     skills = st.text_input("Skills (e.g., Python, Marketing)")
     interests = st.text_input("Interests (e.g., Technology, Science)")
+    target_language = st.selectbox("Select target language", ['de', 'fr', 'es', 'it', 'pt'])  # Available language codes for translation
 
     submit_button = st.form_submit_button("Find Opportunities")
 
@@ -67,9 +77,13 @@ if submit_button:
     if scholarships:
         st.write("Scholarships found:")
         for scholarship in scholarships:
-            st.write(f"Title: {scholarship['title']}")
-            st.write(f"Description: {scholarship.get('description', 'No description available')}")
-            st.write(f"Eligibility: {scholarship.get('eligibility', 'No eligibility available')}")
+            title = translate_text(scholarship.get('title', 'No title available'), target_language)
+            description = translate_text(scholarship.get('description', 'No description available'), target_language)
+            eligibility = translate_text(scholarship.get('eligibility', 'No eligibility available'), target_language)
+            
+            st.write(f"Title: {title}")
+            st.write(f"Description: {description}")
+            st.write(f"Eligibility: {eligibility}")
             st.write("---")
     else:
         st.write("No scholarships found for the selected country.")
@@ -78,10 +92,15 @@ if submit_button:
     if internships:
         st.write("Internships found:")
         for internship in internships:
-            st.write(f"Title: {internship['jobtitle']}")
-            st.write(f"Company: {internship['company']}")
-            st.write(f"Location: {internship['location']}")
-            st.write(f"Snippet: {internship['snippet']}")
+            title = translate_text(internship.get('jobtitle', 'No title available'), target_language)
+            company = translate_text(internship.get('company', 'No company available'), target_language)
+            location = translate_text(internship.get('location', 'No location available'), target_language)
+            snippet = translate_text(internship.get('snippet', 'No snippet available'), target_language)
+            
+            st.write(f"Title: {title}")
+            st.write(f"Company: {company}")
+            st.write(f"Location: {location}")
+            st.write(f"Snippet: {snippet}")
             st.write("---")
     else:
         st.write("No internships found for the selected country.")
@@ -92,7 +111,11 @@ if submit_button:
     
     st.write("AI-based Recommended Opportunities based on your profile:")
     for opportunity in recommended_opportunities:
-        st.write(f"Title: {opportunity['title']}")
-        st.write(f"Description: {opportunity.get('description', 'No description available')}")
-        st.write(f"Eligibility: {opportunity.get('eligibility', 'Not available')}")
+        title = translate_text(opportunity.get('title', 'No title available'), target_language)
+        description = translate_text(opportunity.get('description', 'No description available'), target_language)
+        eligibility = translate_text(opportunity.get('eligibility', 'Not available'), target_language)
+        
+        st.write(f"Title: {title}")
+        st.write(f"Description: {description}")
+        st.write(f"Eligibility: {eligibility}")
         st.write("---")