Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import google.generativeai as genai
+import openai
 import os
 import re  # For improved parsing
 
@@ -7,24 +8,39 @@ import re  # For improved parsing
 def login_page():
     st.title("Quiz Generator Login")
 
-    api_options = ["Gemini API"] # Extend this list for more APIs in the future
+    api_options = ["Gemini API", "OpenAI API"]
     selected_api = st.selectbox("Select API:", api_options)
     api_key = st.text_input(f"Enter your {selected_api} Key:", type="password")
 
+    if selected_api == "Gemini API":
+        model_options = ["gemini-pro"] # Add more Gemini models if available
+        selected_model = st.selectbox("Select Gemini Model:", model_options, index=0) # Default to first model
+    elif selected_api == "OpenAI API":
+        model_options = ["gpt-3.5-turbo", "gpt-4"] # Add more OpenAI models as needed
+        selected_model = st.selectbox("Select OpenAI Model:", model_options, index=0) # Default to first model
+    else:
+        selected_model = None # Should not happen, but for safety
+
     if st.button("Login"):
         if not api_key:
             st.error("Please enter your API key.")
         else:
             st.session_state.api_key = api_key
             st.session_state.selected_api = selected_api
+            if selected_api == "Gemini API":
+                st.session_state.gemini_model = selected_model
+                st.session_state.openai_model = None # Clear OpenAI model if Gemini is selected
+            elif selected_api == "OpenAI API":
+                st.session_state.openai_model = selected_model
+                st.session_state.gemini_model = None # Clear Gemini model if OpenAI is selected
             st.session_state.logged_in = True
-            st.success(f"Logged in with {selected_api}!")
+            st.success(f"Logged in with {selected_api} using model: {selected_model}!")
             st.rerun() # Rerun to show the quiz app
 
 # --- Quiz Application ---
 def quiz_app():
     st.title("Interactive Multiple Choice Quiz Generator")
-    st.markdown("Powered by Gemini API") # Keeping this as title, can be dynamically updated if more APIs are added
+    st.markdown(f"Powered by {st.session_state.selected_api}")
 
     # Security Warning - Moved here as it is part of the quiz app
     st.markdown("<font color='red'>**Warning:** Directly entering your API key is less secure than using Streamlit Secrets or environment variables. For production, use Secrets.</font>", unsafe_allow_html=True)
@@ -45,7 +61,7 @@ def quiz_app():
     if st.session_state.selected_api == "Gemini API":
         try:
             genai.configure(api_key=st.session_state.api_key)
-            model = genai.GenerativeModel('gemini-pro')
+            model = genai.GenerativeModel(st.session_state.gemini_model)
         except Exception as e:
             st.error(f"Error configuring Gemini API: {e}")
             st.error("Please check your API key and network connection.")
@@ -54,6 +70,18 @@ def quiz_app():
             del st.session_state.selected_api
             st.rerun() # Go back to login page
             return # Exit quiz_app
+    elif st.session_state.selected_api == "OpenAI API":
+        try:
+            openai.api_key = st.session_state.api_key
+            model_name = st.session_state.openai_model # Model name already selected in login
+        except Exception as e:
+            st.error(f"Error configuring OpenAI API: {e}")
+            st.error("Please check your API key and network connection.")
+            st.session_state.logged_in = False # Force logout if API config fails
+            del st.session_state.api_key
+            del st.session_state.selected_api
+            st.rerun() # Go back to login page
+            return # Exit quiz_app
 
 
     def parse_quiz_content(quiz_content):
@@ -225,8 +253,15 @@ def quiz_app():
 
                     Please generate the quiz in this exact format.
                     """
-                    response = model.generate_content(prompt)
-                    quiz_content = response.text
+                    if st.session_state.selected_api == "Gemini API":
+                        response = model.generate_content(prompt)
+                        quiz_content = response.text
+                    elif st.session_state.selected_api == "OpenAI API":
+                        response = openai.ChatCompletion.create(
+                            model=st.session_state.openai_model,
+                            messages=[{"role": "user", "content": prompt}]
+                        )
+                        quiz_content = response.choices[0].message.content
 
 
                     parsed_quiz_data, answer_key = parse_quiz_content(quiz_content)