Update app.py

app.py

@@ -1,158 +1,125 @@
 import streamlit as st
 import google.generativeai as genai
 from github import Github
+from pypdf import PdfReader
 import os
 from dotenv import load_dotenv
 
-# --- CONFIG & SETUP ---
-st.set_page_config(page_title="DevResume AI", layout="wide", page_icon="🚀")
+# --- CONFIG ---
+st.set_page_config(page_title="DevResume Pro", layout="wide", page_icon="⚡")
 
-# Load Keys (Cloud or Local)
+# Load Keys
 api_key = os.getenv("GEMINI_API_KEY")
 github_token = os.getenv("GITHUB_TOKEN")
 
-# Fallback for local testing
 if not api_key:
     load_dotenv()
     api_key = os.getenv("GEMINI_API_KEY")
     github_token = os.getenv("GITHUB_TOKEN")
 
-# --- ENGINE 1: GITHUB CRAWLER ---
+# --- FUNC 1: PDF PARSER ---
+def extract_pdf_text(file):
+    try:
+        reader = PdfReader(file)
+        text = ""
+        for page in reader.pages:
+            text += page.extract_text()
+        return text
+    except:
+        return None
+
+# --- FUNC 2: GITHUB CRAWLER ---
 def fetch_github_data(username):
-    """
-    SOTA Scraper: Fetches top repos, stars, languages, and README summaries.
-    """
     try:
-        # Auth guarantees higher rate limits
         g = Github(github_token) if github_token else Github()
         user = g.get_user(username)
+        repos = sorted(user.get_repos(), key=lambda x: x.stargazers_count, reverse=True)[:4]
         
-        data = {
-            "name": user.name or username,
-            "bio": user.bio or "AI & Software Engineer",
-            "url": user.html_url,
-            "repos": []
-        }
-        
-        # Sort repos by Stars (Metric of Impact)
-        repos = user.get_repos()
-        sorted_repos = sorted(repos, key=lambda x: x.stargazers_count, reverse=True)[:6]
-        
-        for repo in sorted_repos:
-            # Skip forked repos to focus on original work
-            if repo.fork:
-                continue
-                
-            # Smart README Fetcher
-            readme_text = "No details provided."
+        repo_data = ""
+        for repo in repos:
+            readme = "No README"
             try:
-                readme = repo.get_readme()
-                if readme.decoded_content:
-                    readme_text = readme.decoded_content.decode("utf-8")[:1500] # Context Window Optimization
-            except:
-                pass
-
-            data["repos"].append({
-                "name": repo.name,
-                "stars": repo.stargazers_count,
-                "lang": repo.language,
-                "desc": repo.description,
-                "readme": readme_text
-            })
+                if repo.get_readme().decoded_content:
+                    readme = repo.get_readme().decoded_content.decode("utf-8")[:1000]
+            except: pass
             
-        return data
+            repo_data += f"Project: {repo.name} ({repo.stargazers_count} Stars, {repo.language})\nDesc: {repo.description}\nContext: {readme}\n\n"
+            
+        return repo_data
     except Exception as e:
-        return {"error": f"GitHub API Error: {str(e)}"}
+        return f"Error fetching GitHub: {str(e)}"
 
-# --- ENGINE 2: GEMINI RESUME ARCHITECT ---
-def generate_resume_content(profile, target_role):
+# --- FUNC 3: THE RESUME ARCHITECT ---
+def generate_full_resume(old_resume_text, github_data, target_role):
     genai.configure(api_key=api_key)
-    
-    # Try SOTA Model first
     try:
         model = genai.GenerativeModel('gemini-2.5-flash')
     except:
         model = genai.GenerativeModel('gemini-1.5-flash')
 
-    # Context Construction
-    repo_context = ""
-    for r in profile['repos']:
-        repo_context += f"""
-        - PROJECT: {r['name']} ({r['stars']} Stars, {r['lang']})
-          DESCRIPTION: {r['desc']}
-          TECHNICAL CONTEXT: {r['readme']}
-        """
-
     prompt = f"""
-    You are a Senior Technical Recruiter for a FAANG company.
+    You are an Elite Tech Resume Writer.
     
-    CANDIDATE: {profile['name']}
-    ROLE: {target_role}
+    SOURCE MATERIAL:
+    1. OLD RESUME TEXT:
+    {old_resume_text[:4000]}
     
-    TASK:
-    Write the "Featured Projects" section of a resume based on their GitHub code.
+    2. GITHUB PORTFOLIO ANALYSIS:
+    {github_data}
     
-    RULES:
-    1. Select the top 3 most technically impressive projects.
-    2. Write 3 bullet points for each project.
-    3. STRICTLY use the "XYZ Pattern" (Accomplished [X] as measured by [Y], by doing [Z]).
-    4. Quantify impact where possible (e.g., "Reduced latency by...", "Processed 10k+ requests...").
-    5. Highlight stack: Python, PyTorch, FastAPI, Docker, etc.
+    TARGET ROLE: {target_role}
     
-    INPUT DATA (GitHub Scrape):
-    {repo_context}
+    TASK:
+    Write a complete, SOTA Resume (Markdown Format).
     
-    OUTPUT FORMAT (Markdown):
-    ## 🚀 Featured Projects
+    STRUCTURE:
+    1. **Header:** Name, Title (Target Role), Links (GitHub/LinkedIn).
+    2. **Professional Summary:** Synthesize the old resume + new GitHub achievements into a killer 3-line bio.
+    3. **Technical Skills:** Grouped by category (Languages, Frameworks, Tools).
+    4. **Featured Projects:** Use the GitHub Data. Write "XYZ Style" impact bullets.
+    5. **Experience:** Polish the old resume's experience section. Make it punchy.
+    6. **Education:** Keep it simple.
     
-    **[Project Name]** | *[Languages/Tools]*
-    * [Bullet 1 using XYZ pattern]
-    * [Bullet 2 using XYZ pattern]
-    * [Bullet 3 using XYZ pattern]
+    TONE: Senior, High-Impact, Quantified.
     """
     
     try:
         response = model.generate_content(prompt)
         return response.text
     except Exception as e:
-        return f"AI Generation Error: {str(e)}"
+        return f"AI Error: {e}"
 
-# --- UI LAYER ---
-with st.sidebar:
-    st.title("🚀 DevResume AI")
-    st.markdown("Turns your **GitHub Code** into a **Senior Resume** instantly.")
-    st.info("Powered by **Gemini 2.5** & **PyGithub**")
-    st.caption("Built by Owadokun Tosin Tobi")
-
-st.title("👨‍💻 GitHub-to-Resume Generator")
-st.markdown("Stop writing generic bullets. Let AI analyze your actual code and write high-impact XYZ statements.")
+# --- UI ---
+st.title("⚡ DevResume Pro: The Hybrid Architect")
+st.markdown("Combines your **Old Resume** + **Real GitHub Code** to build the ultimate portfolio.")
 
 col1, col2 = st.columns([1, 2])
 
 with col1:
+    st.subheader("1. Your Inputs")
     gh_user = st.text_input("GitHub Username", placeholder="eatosin")
     role = st.text_input("Target Role", value="Senior AI Engineer")
-    generate_btn = st.button("✨ Generate Resume", type="primary")
-
-with col2:
-    if generate_btn:
-        if not gh_user:
-            st.warning("Please enter a username.")
+    uploaded_file = st.file_uploader("Upload Old Resume (PDF)", type="pdf")
+    
+    if st.button("🚀 Build Resume", type="primary"):
+        if not gh_user or not uploaded_file:
+            st.error("Please provide both GitHub User and PDF Resume.")
         else:
-            with st.status("🕷️ Crawling GitHub Ecosystem...", expanded=True) as status:
-                st.write("Connecting to GitHub API...")
-                data = fetch_github_data(gh_user)
+            with st.spinner("🕵️‍♂️ Reading PDF & Scraping GitHub..."):
+                # 1. Read PDF
+                pdf_text = extract_pdf_text(uploaded_file)
+                
+                # 2. Scrape GitHub
+                gh_data = fetch_github_data(gh_user)
                 
-                if "error" in data:
-                    status.update(label="❌ Error", state="error")
-                    st.error(data['error'])
+                # 3. Generate
+                if pdf_text and gh_data:
+                    st.session_state['resume'] = generate_full_resume(pdf_text, gh_data, role)
                 else:
-                    st.write(f"✅ Found {len(data['repos'])} repositories.")
-                    st.write("🧠 Analyzing READMEs & Codebase...")
-                    resume = generate_resume_content(data, role)
-                    status.update(label="✅ Resume Generated!", state="complete")
-                    
-                    st.markdown("---")
-                    st.subheader("📄 Generated Project Section")
-                    st.markdown(resume)
-                    st.download_button("📥 Download Markdown", resume, file_name="featured_projects.md")
+                    st.error("Failed to read inputs.")
+
+with col2:
+    if 'resume' in st.session_state:
+        st.subheader("📄 Your New Resume")
+        st.markdown(st.session_state['resume'])
+        st.download_button("📥 Download Markdown", st.session_state['resume'], file_name="SOTA_Resume.md")