Update app.py

app.py

@@ -20,18 +20,23 @@ import os
 
 
 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}."
+    question = f"Generate an interview question for the role of {role} on the topic of {topic} with difficulty level {difficulty_level}."
+    template = """Question: {question}
+                  Answer: 
+                """
     # llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash-lite", google_api_key=st.secrets["GOOGLE_API_KEY"])
-    repo_id = "Qwen/Qwen3-8B"
+    repo_id =  "mistralai/Mistral-7B-Instruct-v0.2" #"Qwen/Qwen3-8B"
     llm = HuggingFaceEndpoint(
         repo_id=repo_id,
         max_length=512,
         temperature=0.5,
     )
     # llm_chain = prompt | llm
-    prompt = PromptTemplate.from_template(prompt)
+    prompt = PromptTemplate.from_template(template)
+    # prompt = PromptTemplate.from_template(prompt)
     llm_chain = prompt | llm
-    response = llm_chain.invoke(prompt)
+    # response = llm_chain.invoke(prompt)
+    llm_chain.invoke({"question": question})
     response = response.content
  
     return response