Update app.py

app.py

@@ -7,9 +7,10 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 
-# Set your Hugging Face token via environment variable or secret management
-os.environ["HUGGINGFACEHUB_API_KEY"] = os.getenv("HF")
-os.environ["HF_TOKEN"] = os.getenv("HF")
+# Handle Hugging Face token
+hf_token = os.getenv("HF")  # For local dev
+if hf_token:
+    os.environ["HUGGINGFACEHUB_API_KEY"] = hf_token
 
 # UI Configuration
 st.set_page_config(page_title="Resume Validator", layout="centered", page_icon="📄")
@@ -26,7 +27,6 @@ uploaded_file = st.file_uploader("📤 Upload Resume (PDF or DOCX)", type=["pdf"
 resume_text = ""
 
 if uploaded_file:
-    # Save the uploaded file temporarily
     with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[-1]) as tmp_file:
         tmp_file.write(uploaded_file.read())
         temp_path = tmp_file.name
@@ -37,11 +37,11 @@ if uploaded_file:
     else:
         resume_text = docx2txt.process(temp_path)
 
-    # Clean up temp file
+    # Remove temp file
     os.remove(temp_path)
 
     st.markdown("### 📃 Extracted Resume Text")
-    st.code(resume_text)
+    st.text_area("Resume Text", resume_text, height=300)
 
     # Prompt template
     template = """
@@ -69,8 +69,8 @@ if uploaded_file:
     llm = HuggingFaceEndpoint(
         repo_id="mistralai/Mistral-7B-Instruct-v0.3",
         temperature=0.5,
-        max_new_tokens=10,
-        task="conversational"
+        max_new_tokens=512,
+        task="text-generation"
     )
 
     model = ChatHuggingFace(llm=llm)
@@ -83,10 +83,8 @@ if uploaded_file:
                 result = chain.run(resume_text=resume_text)
                 st.success("✅ Resume Analysis Completed")
                 st.markdown("### 📊 Feedback")
-                st.write(result)
+                st.markdown(result)
             except Exception as e:
                 st.error(f"⚠️ An error occurred: {e}")
-
 else:
     st.markdown("<center><i>Please upload your resume to start validation.</i></center>", unsafe_allow_html=True)
-