Update app.py (#1)

- Update app.py (67017fe94d1883c6779ba0802f8e63063d9d6e04)

app.py

@@ -131,52 +131,81 @@
 
 
 
-
-
 import openai
 import streamlit as st
-from langchain_community.llms import OpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 
-# Set your OpenAI API key
-openai.api_key = st.secrets["OPENAI_API_KEY"]
 
 def generate_question(role, topic, difficulty_level):
     prompt = f"Generate an interview question for the role of {role} on the topic of {topic} with difficulty level {difficulty_level}."
-    
-    
-    llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"])
-    response = llm(prompt)
-   
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
+    response = llm.invoke(prompt)
+    response = response.content
+ 
     return response
 
 def evaluate_answer(question, user_answer):
     prompt = f"Question: {question}\nUser's Answer: {user_answer}\nEvaluate the answer and provide feedback. Also, provide the best possible answer."
-    llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"])
-    response = llm(prompt)
-    
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
+    response = llm.invoke(prompt)
+    response = response.content
+ 
     return response
 
+# Initialize session state
+if 'questions' not in st.session_state:
+    st.session_state['questions'] = []
+if 'answers' not in st.session_state:
+    st.session_state['answers'] = []
+if 'feedback' not in st.session_state:
+    st.session_state['feedback'] = []
+if 'current_question' not in st.session_state:
+    st.session_state['current_question'] = 0
+if 'total_questions' not in st.session_state:
+    st.session_state['total_questions'] = 10
+
 st.title("Mock Interview Bot")
 
 role = st.selectbox("Select the role:", ["Software Engineer", "Data Scientist", "Product Manager"])
 topic = st.text_input("Enter the topic:")
 difficulty_level = st.selectbox("Select difficulty level:", ["Easy", "Medium", "Hard"])
 
-if st.button("Generate Question"):
+if st.button("Start Interview"):
     if role and topic and difficulty_level:
-        question = generate_question(role, topic, difficulty_level)
-        st.session_state['current_question'] = question
-        st.write(f"Question: {question}")
+        for _ in range(st.session_state['total_questions']):
+            question = generate_question(role, topic, difficulty_level)
+            st.session_state['questions'].append(question)
+        st.session_state['current_question'] = 0
+        st.write(f"Question 1: {st.session_state['questions'][0]}")
         st.session_state['question_answered'] = False
 
-if 'current_question' in st.session_state and not st.session_state.get('question_answered', False):
-    answer = st.text_area("Your Answer:")
-    if st.button("Submit Answer"):
-        if answer:
-            st.session_state['user_answer'] = answer
-            st.session_state['question_answered'] = True
-
-if 'user_answer' in st.session_state:
-    with st.spinner("Evaluating your answer..."):
-        feedback = evaluate_answer(st.session_state['current_question'], st.session_state['user_answer'])
-        st.write(f"Feedback and Best Answer:\n{feedback}")
+if 'questions' in st.session_state and st.session_state['current_question'] < st.session_state['total_questions']:
+    if not st.session_state.get('question_answered', False):
+        answer = st.text_area("Your Answer:")
+        if st.button("Submit Answer"):
+            if answer:
+                current_question = st.session_state['current_question']
+                st.session_state['answers'].append(answer)
+                feedback = evaluate_answer(st.session_state['questions'][current_question], answer)
+                st.session_state['feedback'].append(feedback)
+                st.session_state['question_answered'] = True
+                st.write(f"Feedback: {feedback}")
+
+                if current_question + 1 < st.session_state['total_questions']:
+                    st.session_state['current_question'] += 1
+                    st.session_state['question_answered'] = False
+                    st.write(f"Question {st.session_state['current_question'] + 1}: {st.session_state['questions'][st.session_state['current_question']]}")
+                else:
+                    st.write("Interview Complete! Generating Report...")
+                    generate_report()
+
+def generate_report():
+    st.write("### Interview Report")
+    for i in range(st.session_state['total_questions']):
+        st.write(f"**Question {i+1}:** {st.session_state['questions'][i]}")
+        st.write(f"**Your Answer:** {st.session_state['answers'][i]}")
+        st.write(f"**Feedback:** {st.session_state['feedback'][i]}")
+        st.write("---")
+
+if 'current_question' in st.session_state and st.session_state['current_question'] == st.session_state['total_questions']:
+    generate_report()