Update app.py

app.py

@@ -13,7 +13,7 @@ def initialize_model():
     for key in API_KEYS:
         try:
             genai.configure(api_key=key)
-            model = genai.GenerativeModel("gemini-2.0-flash")
+            model = genai.GenerativeModel("gemini-1.5-flash")
             test = model.generate_content("Hello!")
             if "Hello" in test.text:
                 print(f"[INFO] Using Gemini API key ending with: ...{key[-4:]}")
@@ -24,331 +24,43 @@ def initialize_model():
 
 initialize_model()
 
-# ========== Load Whisper ==========
-whisper_model = whisper.load_model("base")
+# ========== Load Whisper Model ==========
+asr_model = whisper.load_model("base")
 
-# ========== Questions ==========
+# ========== All Questions (Shortened Example) ==========
 all_questions = {
     "general": [
-    "Tell me about yourself.",
-    "What are your strengths and weaknesses?",
-    "How do you handle stress or pressure?",
-    "Describe a time when you worked in a team.",
-    "How do you manage your time and prioritize tasks?",
-    "Why do you want to join this company?",
-    "Where do you see yourself in 5 years?",
-    "Describe a challenge or conflict you faced and how you dealt with it.",
-    "What motivates you to do your best?",
-    "How do you handle feedback and criticism?",
-    "Tell me about a time you took initiative.",
-    "How do you stay organized during busy schedules?",
-    "What do you do when you face a task you don’t know how to solve?",
-    "Tell me about a mistake you made and what you learned from it.",
-    "Have you ever led a project or team? What was your experience?",
-    "How do you manage deadlines when working on multiple tasks?",
-    "Why should we hire you?",
-    "Tell me about a time you failed. How did you respond?",
-    "How do you approach learning something new?",
-    "What do you know about our company?",
-    "How do you balance quality and speed in your work?",
-    "Describe a time when you had to make a difficult decision.",
-    "How do you communicate in a remote or hybrid work environment?",
-    "What kind of work culture are you most comfortable in?",
-    "Do you prefer working independently or in a team? Why?"
-    ],
-    "web development": [
-        "What is the difference between HTML and XHTML?",
-        "Explain how RESTful APIs work.",
-        "What are the advantages of using React over plain JavaScript?",
-        "How does the virtual DOM improve performance?",
-        "What are HTTP methods? Explain each.",
-        "What is the difference between sessionStorage and localStorage?",
-        "What is CORS and how do you handle it?",
-        "Explain the box model in CSS.",
-        "What is responsive design?",
-        "What is the difference between SQL and NoSQL databases?",
-        "What are web sockets and when are they used?",
-        "What is a service worker in PWA?",
-        "What is lazy loading and why is it useful?",
-        "Difference between GET and POST requests?",
-        "How does server-side rendering differ from client-side rendering?",
-        "What is Next.js and how is it different from React?",
-        "What is a CDN and how does it help?",
-        "What is HTTPS and why is it important?",
-        "Explain the concept of middleware in Express.js.",
-        "What is an ORM and why use it?",
-        "Explain MVC architecture.",
-        "What is the difference between PUT and PATCH requests?",
-        "What is GraphQL and how does it compare to REST?",
-        "How do cookies and sessions work?",
-        "What is JWT and how is it used in authentication?"
-    ],
-    "devops": [
-        "What is CI/CD and why is it important?",
-        "Explain the concept of Infrastructure as Code.",
-        "What is Docker and how is it different from a virtual machine?",
-        "What are containers and how do they help in deployment?",
-        "What is Kubernetes used for?",
-        "How do you monitor applications in production?",
-        "What is the role of Jenkins in DevOps?",
-        "What are some popular logging tools?",
-        "Explain blue-green deployment.",
-        "What is Ansible used for?",
-        "What is Terraform and how is it used?",
-        "What is a build pipeline?",
-        "What is Helm in Kubernetes?",
-        "What is a rolling update in deployments?",
-        "Explain canary deployments.",
-        "What is load balancing and how does it work?",
-        "What is the difference between horizontal and vertical scaling?",
-        "What are artifacts in a CI/CD pipeline?",
-        "What is the use of Prometheus and Grafana?",
-        "What is a reverse proxy?",
-        "How do you ensure zero-downtime deployments?",
-        "Explain the role of GitOps.",
-        "What are secrets management tools?",
-        "What is the purpose of using NGINX in DevOps?",
-        "What is a container registry?"
-    ],
-    "cybersecurity": [
-        "What is the difference between symmetric and asymmetric encryption?",
-        "What is SQL injection and how do you prevent it?",
-        "What are some common types of malware?",
-        "What is the CIA triad?",
-        "What is multi-factor authentication?",
-        "What are firewalls and how do they work?",
-        "Explain HTTPS and how it secures data.",
-        "What is penetration testing?",
-        "How do you secure an API?",
-        "What are some best practices for password security?",
-        "What is phishing and how can it be prevented?",
-        "What is social engineering?",
-        "What is a DDoS attack?",
-        "What are intrusion detection systems?",
-        "What is a VPN and how does it work?",
-        "What is data encryption at rest vs in transit?",
-        "What is a zero-day vulnerability?",
-        "Explain role-based access control (RBAC).",
-        "What is a digital certificate?",
-        "How does a public key infrastructure (PKI) work?",
-        "What is a man-in-the-middle attack?",
-        "What are honeypots in cybersecurity?",
-        "Explain sandboxing in malware analysis.",
-        "What are security patches?",
-        "What is OWASP Top 10?"
-    ],
-    "ai": [
-        "What is the difference between AI, ML, and Deep Learning?",
-        "Explain a neural network and how it learns.",
-        "What is gradient descent?",
-        "What is the role of activation functions in neural networks?",
-        "Explain supervised vs. unsupervised learning.",
-        "What is backpropagation?",
-        "What is overfitting and how do you avoid it?",
-        "What is transfer learning?",
-        "What are some common use-cases of AI?",
-        "How do you evaluate a classification model?",
-        "What is reinforcement learning?",
-        "What is an autoencoder?",
-        "Explain convolutional neural networks (CNNs).",
-        "What are recurrent neural networks (RNNs)?",
-        "What is a transformer model?",
-        "What is attention mechanism in NLP?",
-        "What is the Turing test?",
-        "What are generative adversarial networks (GANs)?",
-        "What is the bias-variance tradeoff?",
-        "What is explainable AI (XAI)?",
-        "How do you train a deep learning model?",
-        "What is early stopping in training?",
-        "What are hyperparameters?",
-        "What is a confusion matrix?",
-        "What is the role of regularization in deep learning?"
+        "Tell me about yourself.", "What are your strengths and weaknesses?",
+        "Describe a time you worked in a team.", "Where do you see yourself in 5 years?",
+        "Why should we hire you?", "How do you manage stress and deadlines?",
+        "Tell me about a challenge you overcame.", "What motivates you to perform?",
+        "How do you handle feedback?", "What is your biggest achievement?",
+        # Add up to 25
     ],
     "ml": [
-        "What is supervised learning? Give an example.",
+        "What is overfitting?", "Explain supervised vs unsupervised learning.",
         "What is the difference between classification and regression?",
-        "How does linear regression work?",
-        "What are decision trees and how do they work?",
-        "What is overfitting vs underfitting?",
-        "Explain cross-validation and its importance.",
-        "What is regularization in machine learning?",
-        "What is the difference between bagging and boosting?",
-        "What is the role of a cost function?",
-        "How do you handle missing data in ML datasets?",
-        "What is the k-nearest neighbors (KNN) algorithm?",
-        "What is support vector machine (SVM)?",
-        "What is Naive Bayes algorithm?",
-        "What is logistic regression?",
-        "What is feature scaling and why is it needed?",
-        "What is principal component analysis (PCA)?",
-        "What is model drift?",
-        "How do you tune hyperparameters?",
-        "What is grid search and random search?",
-        "What are ensemble models?",
-        "What is a confusion matrix and what do its values represent?",
-        "How do you balance an imbalanced dataset?",
-        "What is precision-recall tradeoff?",
-        "What is ROC-AUC score?",
-        "What is the purpose of stratified sampling?"
-    ],
-    "mobile development": [
-        "What is the difference between native and hybrid apps?",
-        "Explain the lifecycle of an Android activity.",
-        "What is Flutter and why use it?",
-        "How do you handle data persistence in mobile apps?",
-        "What is MVVM architecture?",
-        "Explain push notifications and how they work.",
-        "What are the limitations of mobile development?",
-        "How do you optimize performance in mobile apps?",
-        "What is the role of App Store Optimization (ASO)?",
-        "Explain the concept of responsive design in mobile.",
-        "What is the difference between Android and iOS development?",
-        "What is React Native and how does it work?",
-        "How do you handle offline mode in apps?",
-        "What is a RecyclerView?",
-        "Explain intents in Android.",
-        "What is dependency injection in Android?",
-        "What are permissions in mobile apps?",
-        "How do you debug a mobile app?",
-        "What is Jetpack Compose?",
-        "How is state management handled in Flutter?",
-        "What is a storyboard in iOS?",
-        "What are background services?",
-        "How do you handle API calls in mobile apps?",
-        "What is the difference between LiveData and Flow?",
-        "How do you publish an app to Google Play Store?"
-    ],
-    "uiux": [
-        "What is the difference between UI and UX?",
-        "What are some principles of good UX design?",
-        "What is a user journey?",
-        "How do you conduct user testing?",
-        "What tools do you use for UI/UX design?",
-        "What is wireframing?",
-        "How do you design for accessibility?",
-        "What is the role of prototyping?",
-        "Explain the concept of design thinking.",
-        "How do you collect and incorporate user feedback?",
-        "What is the importance of consistency in UI design?",
-        "What is heuristic evaluation?",
-        "What is the role of typography in UI?",
-        "What are color theory principles?",
-        "How does UX impact business goals?",
-        "What is A/B testing in UI/UX?",
-        "What is microcopy and why is it important?",
-        "Explain responsive vs adaptive design.",
-        "What is a user persona?",
-        "What is interaction design?",
-        "What is a high-fidelity vs low-fidelity prototype?",
-        "How do you measure UX success?",
-        "What is card sorting?",
-        "What are accessibility standards (WCAG)?",
-        "What is the role of animation in UI?"
-    ],
-    "data analytics": [
-        "What is the difference between descriptive and inferential statistics?",
-        "What are some common data cleaning techniques?",
-        "Explain how to deal with missing data.",
-        "What is the role of data visualization?",
-        "What is a pivot table?",
-        "What are outliers and how do you handle them?",
-        "Explain correlation vs causation.",
-        "What are some commonly used data analytics tools?",
-        "How do you handle large datasets efficiently?",
-        "What is ETL and how does it work?",
-        "What is a histogram and when is it used?",
-        "What is data wrangling?",
-        "What is the difference between mean, median, and mode?",
-        "What is standard deviation and variance?",
-        "What is a box plot?",
-        "What is a time series analysis?",
-        "What is the difference between R and Python for analytics?",
-        "What is the use of SQL in data analytics?",
-        "How do dashboards help in analytics?",
-        "What is cohort analysis?",
-        "What is hypothesis testing?",
-        "What are confidence intervals?",
-        "How do you avoid bias in analytics?",
-        "What is the purpose of a heatmap?",
-        "What is a funnel analysis?"
-    ],
-    "data scientist": [
-        "What is the difference between supervised and unsupervised learning?",
-        "Explain bias-variance tradeoff in machine learning.",
-        "How do you handle imbalanced datasets?",
-        "What are precision, recall, and F1-score? Why are they important?",
-        "Describe a machine learning project you've worked on.",
-        "What is feature engineering? Give examples.",
-        "How do you select important variables in a dataset?",
-        "Explain overfitting and how to prevent it.",
-        "What’s the difference between bagging and boosting?",
-        "Describe the steps in a typical machine learning pipeline.",
-        "What tools do you use as a data scientist?",
-        "What is dimensionality reduction?",
-        "How do you evaluate regression models?",
-        "What is the difference between ROC and PR curves?",
-        "What is model interpretability?",
-        "How do you deploy a machine learning model?",
-        "What is a recommendation system?",
-        "How do you deal with outliers?",
-        "What is SMOTE?",
-        "What is A/B testing in data science?",
-        "How do you build a data pipeline?",
-        "What are data lakes and data warehouses?",
-        "What is a baseline model?",
-        "How do you perform EDA (Exploratory Data Analysis)?",
-        "What is clustering and what are its applications?"
-    ],
-    "prompt engineering": [
-        "What is prompt engineering and why is it important?",
-        "How do temperature and top_p affect model output in LLMs?",
-        "How do you write a good system prompt for a chatbot?",
-        "What are the risks of prompt injection attacks?",
-        "What is chain-of-thought prompting?",
-        "How does few-shot prompting improve performance?",
-        "What are embeddings, and how are they used in retrieval-augmented generation?",
-        "What are the limitations of prompt-based techniques?",
-        "Explain the role of role-based prompts in function calling.",
-        "How would you structure a prompt to summarize a large document accurately?",
-        "What is zero-shot learning in prompt design?",
-        "What is in-context learning?",
-        "What is semantic search and how is it used with LLMs?",
-        "How do you evaluate the output of an LLM?",
-        "What is hallucination in LLMs?",
-        "What are guardrails in LLM applications?",
-        "What is prompt chaining?",
-        "How do you personalize prompts for users?",
-        "What are system vs user prompts?",
-        "What is instruction tuning?",
-        "How do you manage memory in prompt-based agents?",
-        "What is RAG (retrieval-augmented generation)?",
-        "What is the difference between deterministic and probabilistic LLM output?",
-        "How do prompts interact with tools in agent systems?",
-        "How do you test prompt robustness?"
+        "How does cross-validation help in model evaluation?",
+        "What is the purpose of regularization?",
+        "Explain decision trees.", "What is PCA?",
+        "What are ensemble methods?", "What is model drift?",
+        "How do you evaluate a regression model?",
+        # Add up to 25
     ]
 }
 
 # ========== Session State ==========
 session = {
-    "questions": [],
-    "current_index": 0,
-    "scores": [],
-    "feedbacks": [],
-    "transcripts": []
+    "questions": [], "current_index": 0,
+    "scores": [], "feedbacks": [], "transcripts": []
 }
 
-# ========== Handle Audio ==========
-def handle_audio(audio):
-    if not audio:
-        return "No audio provided"
-    result = whisper_model.transcribe(audio)
+# ========== Transcribe Audio ==========
+def transcribe_audio(audio):
+    result = asr_model.transcribe(audio)
     return result["text"]
 
-# Auto-transcription for transcript box
-def transcribe_and_show(audio):
-    return handle_audio(audio)
-
-# ========== Evaluate Answer ==========
+# ========== Get Score From Gemini ==========
 def get_score(question, answer):
     prompt = f"""
 You are an interview expert. A user was asked:
@@ -361,85 +73,65 @@ Format:
 1. **Rating:** <score>/10
 2. **Feedback:** <text>
 """
-    response = model.generate_content(prompt).text.strip()
-    score = -1
-    feedback = "Feedback not found"
-    for line in response.split("\n"):
+    res = model.generate_content(prompt).text
+    score = 0
+    feedback = "No feedback"
+    for line in res.splitlines():
         if "**Rating:**" in line:
-            try:
-                score = int(line.split("**Rating:**")[1].split("/")[0].strip())
-            except:
-                pass
-        elif "**Feedback:**" in line:
-            feedback = line.split("**Feedback:**")[1].strip()
+            score = int(line.split("**Rating:**")[-1].split("/")[0].strip())
+        if "**Feedback:**" in line:
+            feedback = line.split("**Feedback:**")[-1].strip()
     return score, feedback
 
-# ========== Summary ==========
-def summarize_feedback(feedbacks):
-    prompt = "Summarize the following feedback into 4-5 improvement tips:\n" + "\n".join(feedbacks)
-    return model.generate_content(prompt).text.strip()
-
-# ========== Submit Answer ==========
-def run_session(audio):
-    if session["current_index"] >= len(session["questions"]):
-        return "✅ Interview complete!", "", gr.update(visible=False)
+# ========== Start Interview ==========
+def start_session(field):
+    general_qs = random.sample(all_questions["general"], 10)
+    domain_qs = random.sample(all_questions[field], 10)
+    session["questions"] = random.sample(general_qs + domain_qs, 20)
+    session["current_index"] = 0
+    session["scores"] = []
+    session["feedbacks"] = []
+    session["transcripts"] = []
+    return session["questions"][0], gr.update(visible=True), ""
 
-    question = session["questions"][session["current_index"]]
-    transcript = handle_audio(audio)
-    session["transcripts"].append(transcript)
+# ========== Process Answer ==========
+def process(audio):
+    idx = session["current_index"]
+    if idx >= len(session["questions"]):
+        return "Interview complete!", "", gr.update(visible=False)
 
+    question = session["questions"][idx]
+    transcript = transcribe_audio(audio)
     score, feedback = get_score(question, transcript)
+
+    session["transcripts"].append(transcript)
     session["scores"].append(score)
-    session["feedbacks"].append(f"Q{session['current_index']+1}: {feedback}")
+    session["feedbacks"].append(f"Q{idx+1}: {feedback}")
     session["current_index"] += 1
 
     if session["current_index"] < len(session["questions"]):
         next_q = session["questions"][session["current_index"]]
-        return (
-            f"🗣️ Answer: {transcript}\n\n🔢 Score: {score}/10\n💬 Feedback: {feedback}\n\n➡️ Next: {next_q}",
-            "",
-            gr.update(visible=True),
-        )
+        return f"🗣 {transcript}\n\n🔢 Score: {score}/10\n💬 {feedback}", next_q, gr.update(visible=True)
     else:
-        avg_score = sum(session["scores"]) / len(session["scores"])
-        summary = summarize_feedback(session["feedbacks"])
-        return (
-            f"✅ Interview Complete!\n\n📊 Avg Score: {avg_score:.2f}/10\n\n📝 Summary:\n{summary}",
-            "",
-            gr.update(visible=False),
-        )
-
-# ========== Start New Session ==========
-def start_session(field):
-    general_qs = random.sample(all_questions["general"], 10)
-    domain_qs = random.sample(all_questions[field], 10)
-    session["questions"] = general_qs + domain_qs
-    random.shuffle(session["questions"])
-    session["current_index"] = 0
-    session["scores"] = []
-    session["feedbacks"] = []
-    session["transcripts"] = []
-    return session["questions"][0], gr.update(visible=True)
+        avg = sum(session["scores"]) / len(session["scores"])
+        return f"✅ Interview Done!\n\nAvg Score: {avg:.2f}/10", "", gr.update(visible=False)
 
 # ========== Gradio UI ==========
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎤 AI Mock Interview (Voice-Based)")
-    with gr.Row():
-        field = gr.Dropdown(
-            label="Select your domain",
-            choices=list(all_questions.keys()),
-            value="ml"
-        )
-        start_btn = gr.Button("Start Interview")
+    gr.Markdown("# 🎤 AI Mock Interview (Voice-Based)")
+
+    with gr.Tab("Start Interview"):
+        field = gr.Dropdown(choices=list(all_questions.keys()), label="Choose your field", value="ml")
+        start_btn = gr.Button("Start")
+        question_box = gr.Textbox(label="Current Question", interactive=False)
 
-    question_display = gr.Textbox(label="Current Question", interactive=False)
-    audio_input = gr.Audio(type="filepath", label="🎙️ Record Your Answer")
-    transcript_display = gr.Textbox(label="📝 Transcript", interactive=False)
-    submit_btn = gr.Button("Submit Answer", visible=False)
-    output = gr.Textbox(label="Feedback", lines=10)
+    with gr.Tab("Interview"):
+        audio_in = gr.Audio(source="microphone", type="filepath", label="Speak your answer")
+        submit_btn = gr.Button("Submit Answer", visible=False)
+        result_box = gr.Textbox(label="Evaluation", lines=6)
 
-    start_btn.click(start_session, inputs=field, outputs=[question_display, submit_btn])
-    audio_input.change(transcribe_and_show, inputs=audio_input, outputs=transcript_display)
-    submit_btn.click(run_session, inputs=audio_input, outputs=[output, transcript_display, submit_btn])
+    start_btn.click(start_session, inputs=field, outputs=[question_box, submit_btn, result_box])
+    submit_btn.click(process, inputs=audio_in, outputs=[result_box, question_box, submit_btn])
 
+# ========== Launch ==========
 demo.launch()