Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import os
-from huggingface_hub import InferenceClient
+from openai import OpenAI
 
-client = InferenceClient(
-    model="HuggingFaceH4/zephyr-7b-beta",
-    token=os.environ.get("Rejected_tk")
+client = OpenAI(
+    base_url="https://router.huggingface.co/novita/v1",
+    api_key=os.environ.get("Rejected_tk"),
 )
 
 SYSTEM_PROMPT = """You are a brutally honest but helpful senior hiring manager with 15 years of experience.
@@ -43,27 +43,17 @@ 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"""Resume:
----
-{resume_text}
----
-Job Description:
----
-{job_description}
----
-Tell them exactly why they will be rejected and give the full structured report."""
-
     try:
-        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,
+        response = client.chat.completions.create(
+            model="meta-llama/llama-3.1-8b-instruct",
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": f"Resume:\n{resume_text}\n\nJob Description:\n{job_description}\n\nGive the full rejection report."}
+            ],
+            max_tokens=1024,
             temperature=0.7,
-            stream=True,
-        ):
-            result += token
-        return result
+        )
+        return response.choices[0].message.content
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
@@ -100,10 +90,8 @@ body { background: #0f0f1a !important; }
 #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; }
+.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; }
+.gr-textbox textarea { background: #1a1a2e !important; color: #e0e0e0 !important; border: 1px solid #333 !important; border-radius: 10px !important; }
 label { color: #ccc !important; font-weight: 600 !important; }
 """
 
@@ -112,38 +100,22 @@ with gr.Blocks(title="Rejected Before Applying", css=css) as demo:
     <div id='title-block'>
         <h1>💔 Rejected Before Applying</h1>
         <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>
+        <p style='color:#666; font-size:0.85em;'>⚡ Powered by Llama-3.1-8B &nbsp;•&nbsp; 🔒 No data stored &nbsp;•&nbsp; 🎯 Brutal honesty guaranteed</p>
     </div>
     """)
 
     with gr.Row(equal_height=True):
         with gr.Column():
-            resume_input = gr.Textbox(
-                label="📄 Your Resume",
-                placeholder="Paste your resume here — skills, experience, education, projects...",
-                lines=16,
-                value=EXAMPLE_RESUME
-            )
+            resume_input = gr.Textbox(label="📄 Your Resume", placeholder="Paste your resume here...", lines=16, value=EXAMPLE_RESUME)
         with gr.Column():
-            jd_input = gr.Textbox(
-                label="💼 Job Description",
-                placeholder="Paste the full job description here...",
-                lines=16,
-                value=EXAMPLE_JD
-            )
+            jd_input = gr.Textbox(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")
-
     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", value="*Your rejection report will appear here...*")
+    output = gr.Markdown(value="*Your rejection report will appear here...*")
 
     analyze_btn.click(fn=analyze, inputs=[resume_input, jd_input], outputs=output)
 
-    gr.HTML("""
-    <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>
-    """)
+    gr.HTML("<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>")
 
 demo.launch()