Update app.py

app.py

@@ -1,71 +1,69 @@
 import streamlit as st
 from langchain.chat_models import ChatOpenAI
-from langchain.prompts import PromptTemplate
+from langchain.prompts import ChatPromptTemplate
+from langchain.schema.runnable import RunnableLambda
 from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
 from PyPDF2 import PdfReader
 import os
 
 # Streamlit UI setup
 st.set_page_config(page_title="AI Job Interview Prep Tool")
-st.title("🤖 AI-Powered Job Interview Preparation")
+st.title("AI-Powered Job Interview Preparation")
 
-# Secure input for OpenAI API Key
+# OpenAI API Key input
 openai_key = st.text_input("🔐 Enter your OpenAI API Key", type="password")
+if not openai_key:
+    st.warning("Please enter your OpenAI API Key to continue.")
+    st.stop()
+
+os.environ["OPENAI_API_KEY"] = openai_key
 
 # User Inputs
-company = st.text_input("🏢 Company Name")
-profile = st.text_input("💼 Profile Name")
-jd = st.text_area("📄 Job Description")
-resume_file = st.file_uploader("📎 Upload Resume (Optional)", type=["pdf"])
+company = st.text_input("Company Name")
+profile = st.text_input("Profile Name")
+jd = st.text_area("Job Description")
+resume_file = st.file_uploader("Upload Resume (Optional)", type=["pdf"])
 
-# Extract text from PDF if resume is uploaded
 resume_text = ""
 if resume_file is not None:
-    pdf = PdfReader(resume_file)
-    for page in pdf.pages:
+    pdf_reader = PdfReader(resume_file)
+    for page in pdf_reader.pages:
         resume_text += page.extract_text() or ""
 
-# Submit button
-submit = st.button("🚀 Generate Interview Questions")
-
-if submit:
-    if not openai_key or not company or not profile or not jd:
-        st.warning("Please fill in all required fields including the OpenAI API Key.")
-    else:
-        # Set environment variable for OpenAI
-        os.environ["OPENAI_API_KEY"] = openai_key
-
-        # Set up OpenAI LLM
-        llm = ChatOpenAI(temperature=0.7, model_name="gpt-4", openai_api_key=openai_key)
+# Prompt Template
+template = PromptTemplate(
+    input_variables=["company", "profile", "jd", "resume"],
+    template="""
+You are an AI interview coach.
+Generate:
+- 2 technical rounds with 7-10 questions each
+    - Each technical round should have:
+        - Conceptual questions
+        - 2-3 hands-on tasks (code-oriented)
+        - 1 case-study type question
+- 1 business + HR round with 7-10 questions
 
-        # Prompt Template
-        base_prompt = PromptTemplate(
-            input_variables=["company", "profile", "jd", "resume"],
-            template="""
-You are an expert HR and Technical Interviewer.
-Given the following details:
+Base the questions on the job description, company, profile and resume.
 Company: {company}
 Profile: {profile}
 Job Description: {jd}
-Resume (if provided): {resume}
+Resume (optional): {resume}
 
-Generate a structured set of interview questions:
-- Round 1: Technical Interview (7-10 questions including 2 hands-on tasks)
-- Round 2: Technical + Case Study (7-10 questions including at least 1 scenario or case-study based question)
-- Round 3: Business and HR Round (7-10 questions focused on communication, decision making, and company alignment)
-
-Present the questions grouped by round in clear bullet format using Markdown.
+Provide questions clearly labeled under each round.
 """
-        )
+)
+
+llm = ChatOpenAI(temperature=0.7)
+chain = LLMChain(llm=llm, prompt=template)
 
-        chain = LLMChain(llm=llm, prompt=base_prompt)
+if st.button("Generate Interview Rounds"):
+    with st.spinner("Generating questions..."):
         response = chain.run({
             "company": company,
             "profile": profile,
             "jd": jd,
-            "resume": resume_text or "Not Provided"
+            "resume": resume_text,
         })
-
-        st.subheader("📌 Suggested Interview Questions")
-        st.markdown(response)
-
+        st.subheader("📋 Interview Questions")
+        st.write(response)