Update app.py

app.py

@@ -104,6 +104,45 @@ def update_question(mock_index, score, incorrect_questions):
     question, ans1, ans2, ans3, ans4, correct = display_mock_question(mock_index, score, incorrect_questions)
     return gr.update(value=question), gr.update(value=ans1, visible=True), gr.update(value=ans2, visible=True), gr.update(value=ans3, visible=True), gr.update(value=ans4, visible=True)
 
+with open("aa2.txt", "r") as f:
+  prompt = f.read()
+
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+    
+# Create the model configuration
+generation_config = {
+    "temperature": 0.5,
+    "top_p": 0.98,
+    "top_k": 64,
+    "max_output_tokens": 8192,
+    "response_mime_type": "text/plain",
+}
+
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-flash-8b-exp-0827",
+    generation_config=generation_config,
+)
+
+# AI Chatbot function
+def chat_with_ai(user_input):
+    chat_session = model.start_chat(
+      history=[
+        {
+          "role": "user",
+          "parts": [ prompt ],
+        },
+        {
+          "role": "model",
+          "parts": [
+            "Sure, I can answer your question. \n",
+          ],
+        },
+      ]
+    )
+    
+    response = chat_session.send_message(user_input)
+    return response.text
+
 # Gradio interface
 with gr.Blocks() as demo:
     available_weeks = get_available_weeks()
@@ -126,6 +165,13 @@ with gr.Blocks() as demo:
                     prev_btn = gr.Button("Previous")
                     reveal_btn = gr.Button("Reveal Answer")
                     next_btn = gr.Button("Next")
+
+            with gr.Column(visible=True):
+                chat_input = gr.Textbox(label="Ask a question to the A.I.")
+                with gr.Row():
+                    gr.Markdown(">Note: This A.I. has only been given the NBAA Management guide. Use the flashcards for specific week information.")
+                chat_output = gr.Markdown(label="AI Response")
+                chat_button = gr.Button("Ask AI")
             
             # Start button action to reveal the flashcard section
             start_button.click(lambda: gr.update(visible=True), inputs=[], outputs=flashcard_section)