Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
 import os
+from huggingface_hub import InferenceClient
 
 client = InferenceClient(
-    model="mistralai/Mistral-7B-Instruct-v0.3",
-    token=os.environ.get("Rejected_tk")  # Uses the Space's built-in HF token automatically
+    model="HuggingFaceH4/zephyr-7b-beta",
+    token=os.environ.get("Rejected_tk")
 )
 
 SYSTEM_PROMPT = """You are a brutally honest but helpful senior hiring manager with 15 years of experience.
@@ -37,37 +37,36 @@ You must output your response in this exact structure:
 ## 💡 One Honest Verdict
 [One sentence: should they apply now, in 3 months, or pivot entirely?]
 
-Be specific. Name actual technologies. Do not be vague. Do not be encouraging unless there is a real reason to be."""
+Be specific. Name actual technologies. Do not be vague."""
 
 
 def analyze(resume_text, job_description):
     if not resume_text.strip() or not job_description.strip():
         return "⚠️ Please paste both your resume and the job description."
 
-    user_message = f"""Here is the candidate's resume:
+    user_message = f"""Resume:
 ---
 {resume_text}
 ---
-
-Here is the job description they want to apply to:
+Job Description:
 ---
 {job_description}
 ---
-
-Tell them exactly why they will be rejected and what to do about it."""
+Tell them exactly why they will be rejected and give the full structured report."""
 
     try:
-        response = client.chat_completion(
-            messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": user_message}
-            ],
-            max_tokens=1024,
+        result = ""
+        for token in client.text_generation(
+            prompt=f"<|system|>\n{SYSTEM_PROMPT}</s>\n<|user|>\n{user_message}</s>\n<|assistant|>\n",
+            max_new_tokens=1024,
             temperature=0.7,
-        )
-        return response.choices[0].message.content
+            stream=True,
+        ):
+            result += token
+        return result
     except Exception as e:
-        return f"❌ Error: {str(e)}\n\nTry refreshing the page and clicking the button again."
+        return f"❌ Error: {str(e)}"
+
 
 EXAMPLE_RESUME = """Name: Priya Sharma
 Education: M.S. Computer Science, 2024
@@ -95,54 +94,55 @@ Requirements:
 - Experience with data pipelines: Spark, Kafka, Airflow
 - Nice to have: ONNX optimization, TensorRT, model quantization"""
 
-
-with gr.Blocks(
-    theme=gr.themes.Soft(),
-    title="Rejected Before Applying",
-    css="""
-    .header { text-align: center; margin-bottom: 20px; }
-    .warning-box { background: #fff3cd; border-left: 4px solid #ff6b35; padding: 12px; border-radius: 6px; }
-    """
-) as demo:
+css = """
+body { background: #0f0f1a !important; }
+.gradio-container { max-width: 1100px !important; margin: auto; }
+#title-block { text-align: center; padding: 30px 0 10px 0; }
+#title-block h1 { font-size: 2.8em; font-weight: 900; background: linear-gradient(90deg, #ff6b6b, #ff8e53, #ff6bff); -webkit-background-clip: text; -webkit-text-fill-color: transparent; }
+#title-block p { color: #aaa; font-size: 1.05em; }
+.gr-button-primary { background: linear-gradient(90deg, #ff416c, #ff4b2b) !important; border: none !important; font-size: 1.1em !important; padding: 14px !important; border-radius: 10px !important; font-weight: 700 !important; letter-spacing: 0.5px; }
+.gr-button-primary:hover { transform: scale(1.02); box-shadow: 0 0 20px #ff416c88 !important; }
+.gr-textbox textarea { background: #1a1a2e !important; color: #e0e0e0 !important; border: 1px solid #333 !important; border-radius: 10px !important; font-size: 0.92em !important; }
+.gr-markdown { background: #1a1a2e !important; border-radius: 12px !important; padding: 20px !important; border: 1px solid #2a2a4a !important; color: #e0e0e0 !important; }
+label { color: #ccc !important; font-weight: 600 !important; }
+"""
+
+with gr.Blocks(title="Rejected Before Applying", css=css) as demo:
     gr.HTML("""
-    <div class='header'>
+    <div id='title-block'>
         <h1>💔 Rejected Before Applying</h1>
-        <p style='font-size:1.1em; color:#666;'>
-            Find out <b>exactly why</b> your resume won't make it — before you waste the application.
-        </p>
-        <p style='color:#888; font-size:0.9em;'>Powered by Qwen2.5-3B • No data stored • Brutal honesty guaranteed</p>
+        <p>Find out <b>exactly why</b> your resume won't make it — before you waste the application.</p>
+        <p style='color:#666; font-size:0.85em;'>⚡ Powered by Zephyr-7B &nbsp;•&nbsp; 🔒 No data stored &nbsp;•&nbsp; 🎯 Brutal honesty guaranteed</p>
     </div>
     """)
 
-    with gr.Row():
+    with gr.Row(equal_height=True):
         with gr.Column():
             resume_input = gr.Textbox(
-                label="📄 Your Resume (paste as plain text)",
+                label="📄 Your Resume",
                 placeholder="Paste your resume here — skills, experience, education, projects...",
-                lines=15,
+                lines=16,
                 value=EXAMPLE_RESUME
             )
         with gr.Column():
             jd_input = gr.Textbox(
-                label="💼 Job Description (paste the full JD)",
-                placeholder="Paste the job description here...",
-                lines=15,
+                label="💼 Job Description",
+                placeholder="Paste the full job description here...",
+                lines=16,
                 value=EXAMPLE_JD
             )
 
-    analyze_btn = gr.Button("🔍 Analyse My Rejection", variant="primary", size="lg")
+    analyze_btn = gr.Button("🔍  Analyse My Rejection", variant="primary", size="lg")
+
+    gr.HTML("<div style='margin:10px 0 6px 0; color:#888; font-size:0.85em; text-align:center;'>⏳ Analysis takes ~15 seconds — hang tight</div>")
 
-    output = gr.Markdown(label="Your Rejection Report")
+    output = gr.Markdown(label="📋 Your Rejection Report", value="*Your rejection report will appear here...*")
 
-    analyze_btn.click(
-        fn=analyze,
-        inputs=[resume_input, jd_input],
-        outputs=output
-    )
+    analyze_btn.click(fn=analyze, inputs=[resume_input, jd_input], outputs=output)
 
     gr.HTML("""
-    <div style='text-align:center; margin-top:20px; color:#aaa; font-size:0.85em;'>
-        Built for the 🤗 Build Small Hackathon • Small model, real talk
+    <div style='text-align:center; margin-top:24px; color:#555; font-size:0.82em;'>
+        Built for the 🤗 <b>Build Small Hackathon</b> &nbsp;•&nbsp; Small model, real talk
     </div>
     """)