Update app.py

app.py

@@ -1,19 +1,71 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-Created on Sun Jul 13 00:02:29 2025
+import streamlit as st
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+from PyPDF2 import PdfReader
+import os
 
-@author: yogita
-"""
+# Streamlit UI setup
+st.set_page_config(page_title="AI Job Interview Prep Tool")
+st.title("🤖 AI-Powered Job Interview Preparation")
+
+# Secure input for OpenAI API Key
+openai_key = st.text_input("🔐 Enter your OpenAI API Key", type="password")
+
+# 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"])
+
+# 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:
+        resume_text += page.extract_text() or ""
 
-import streamlit as st 
+# Submit button
+submit = st.button("🚀 Generate Interview Questions")
 
-st.title('Basic Addition Calculator') 
+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
 
-number1 = st.number_input("Enter Number 1", min_value=0, max_value=100, value=12) 
-number2 = st.number_input("Enter Number 2", min_value=0, max_value=100, value=12) 
+        # Set up OpenAI LLM
+        llm = ChatOpenAI(temperature=0.7, model_name="gpt-4", openai_api_key=openai_key)
+
+        # Prompt Template
+        base_prompt = PromptTemplate(
+            input_variables=["company", "profile", "jd", "resume"],
+            template="""
+You are an expert HR and Technical Interviewer.
+Given the following details:
+Company: {company}
+Profile: {profile}
+Job Description: {jd}
+Resume (if provided): {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.
+"""
+        )
 
+        chain = LLMChain(llm=llm, prompt=base_prompt)
+        response = chain.run({
+            "company": company,
+            "profile": profile,
+            "jd": jd,
+            "resume": resume_text or "Not Provided"
+        })
 
-number3 = number1 + number2
+        st.subheader("📌 Suggested Interview Questions")
+        st.markdown(response)
 
-st.markdown(number3)