Create app.py

app.py

@@ -0,0 +1,54 @@
+import requests
+from bs4 import BeautifulSoup
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+import gradio as gr
+
+# Function to scrape LinkedIn profile
+def scrape_linkedin_profile(url):
+    response = requests.get(url)
+    if response.status_code == 200:
+        soup = BeautifulSoup(response.content, 'html.parser')
+        profile = {}
+
+        # Example scraping logic (adjust based on actual LinkedIn page structure)
+        profile['name'] = soup.find('title').text.strip()
+        profile['headline'] = soup.find('div', {'class': 'ph5'}).text.strip()
+        profile['about'] = soup.find('section', {'id': 'about'}).text.strip() if soup.find('section', {'id': 'about'}) else "No About Section"
+
+        return profile
+    else:
+        return f"Failed to fetch LinkedIn page, status code: {response.status_code}"
+
+# Function to generate roast using a Gen AI model
+def generate_roast(profile_data):
+    # Initialize the Gen AI model from Hugging Face
+    tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
+    model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
+    
+    # Format profile into a prompt
+    prompt = f"Roast this LinkedIn profile:\nName: {profile_data['name']}\nHeadline: {profile_data['headline']}\nAbout: {profile_data['about']}\n\nRoast:"
+    
+    generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
+    
+    # Generate roast
+    roast = generator(prompt, max_length=150, num_return_sequences=1)
+    
+    return roast[0]['generated_text']
+
+# Gradio interface function
+def roast_linkedin(url):
+    profile_data = scrape_linkedin_profile(url)
+    
+    if isinstance(profile_data, dict):
+        roast = generate_roast(profile_data)
+        return roast
+    else:
+        return profile_data
+
+# Create Gradio interface
+interface = gr.Interface(fn=roast_linkedin, inputs="text", outputs="text", 
+                         title="LinkedIn Profile Roaster",
+                         description="Enter the LinkedIn profile URL and get a humorous roast generated by AI!")
+
+# Launch Gradio app
+interface.launch()