Update app.py

app.py

@@ -1,34 +1,70 @@
-# -*- coding: utf-8 -*-
-"""
-Fancy Addition Calculator with Sidebar and Animations
-Created on Sat Jul 12 10:49:24 2025
-@author: somil
-"""
-
 import streamlit as st
-import time
+from langchain.chat_models import ChatOpenAI
+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")
+
+# It's good practice to load these from a .env file, but for demonstration purposes, we are setting them directly
+os.environ["OPENAI_API_KEY"] = "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"  # Replace with your actual OpenAI API 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"])
+
+# 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()
 
-# Page Config
-st.set_page_config(page_title="Addition Calculator for Eveeryoneskjbgk;jasbgkj;asbga", page_icon="➕", layout="centered")
+# Submit button
+submit = st.button("Generate Interview Questions")
 
-# Sidebar for input
-st.sidebar.header("Enter Numbers Here")
-number1 = st.sidebar.number_input("Enter Number 1", min_value=0, max_value=100, value=12)
-number2 = st.sidebar.number_input("Enter Number 2", min_value=0, max_value=100, value=12)
+if submit:
+    if not company or not profile or not jd:
+        st.warning("Please fill in all mandatory fields.")
+    else:
+        # Set up OpenAI LLM
+        llm = ChatOpenAI(temperature=0.7, model_name="gpt-4")
 
-# Main title and animation
-st.markdown("<h1 style='text-align: center;'>✨ Basic Addition Calculator for Everyone!!!!!✨</h1>", unsafe_allow_html=True)
-st.markdown("<h3 style='text-align: center;'>Let’s crunch some numbers ➕➕</h3>", unsafe_allow_html=True)
+        # 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}
 
-# Animation for effect
-with st.spinner("Calculating... 🧠"):
-    time.sleep(1)  # Simple delay for effect
+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)
 
-# Result
-result = number1 + number2
+Present the questions grouped by round in clear bullet format.
+            """
+        )
 
-st.success(f"🎉 The sum of **{number1}** and **{number2}** is: **{result}**")
+        chain = LLMChain(llm=llm, prompt=base_prompt)
+        response = chain.run({
+            "company": company,
+            "profile": profile,
+            "jd": jd,
+            "resume": resume_text or "Not Provided"
+        })
 
-# Footer message
-st.markdown("---")
-st.markdown("<p style='text-align: center;'>🌟 Hope you have a <b>nice</b> day! 🌈</p>", unsafe_allow_html=True)
+        st.subheader("Suggested Interview Questions")
+        st.write(response)