Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import json
 import os
+import google.generativeai as genai
 
 # Function to get available JSON files in the working directory
 def get_available_weeks():
@@ -40,6 +41,45 @@ def change_question(week_file, index, direction):
     index = (index + direction) % total
     return index, *flashcard_ui(week_file, index)
 
+with open("aa2.txt", "r") as f:
+  prompt = f.read()
+
+# AI Chatbot function
+def chat_with_ai(user_input):
+    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",
+        generation_config=generation_config,
+    )
+    
+    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()
@@ -59,6 +99,12 @@ with gr.Blocks() as demo:
             prev_btn = gr.Button("Previous")
             reveal_btn = gr.Button("Reveal Answer")
             next_btn = gr.Button("Next")
+    
+    # AI Chatbot UI
+    with gr.Column(visible=True):
+        chat_input = gr.Textbox(label="Ask a question to the AI")
+        chat_output = gr.Textbox(label="AI Response", interactive=False)
+        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)
@@ -69,5 +115,8 @@ with gr.Blocks() as demo:
     prev_btn.click(change_question, inputs=[week_file, index, gr.Number(-1, visible=False)], outputs=[index, question_output, answer_output])
     next_btn.click(change_question, inputs=[week_file, index, gr.Number(1, visible=False)], outputs=[index, question_output, answer_output])
 
+    # Chatbot button action
+    chat_button.click(chat_with_ai, inputs=[chat_input], outputs=[chat_output])
+
 # Launch the app
 demo.launch()