Update app.py

app.py

@@ -1,78 +1,57 @@
 import streamlit as st
-import os
 from google import genai
+from google.genai import types
+import json
 from PIL import Image
-import io
-import base64
 
 # Page config
-st.set_page_config(page_title="Exercise Solver", layout="wide")
+st.set_page_config(page_title="Gemini API Physics Questions", layout="wide")
 
-# Initialize Gemini client
-GOOGLE_API_KEY = os.environ.get("GEMINI_API_KEY")
-client = genai.Client(
-    api_key=GOOGLE_API_KEY,
-    http_options={'api_version': 'v1alpha'},
-)
+# Title
+st.title("AP Physics C Practice Question Generator")
 
-def process_image(image):
-    # Convert image to base64
-    buffered = io.BytesIO()
-    image.save(buffered, format="PNG")
-    img_str = base64.b64encode(buffered.getvalue()).decode()
+# Initialize the API key input
+api_key = st.text_input("Enter your Google API Key", type="password")
+
+if api_key:
+    # Initialize the client
+    client = genai.Client(
+        api_key=api_key,
+        http_options={'api_version': 'v1alpha'},
+    )
     
-    # Initialize containers for thoughts and answers
-    thoughts = []
-    answers = []
+    model_name = "gemini-2.0-flash-thinking-exp-01-21"
     
-    try:
+    if st.button("Generate New Question"):
+        # Create containers for thinking and answer
+        thinking_container = st.container()
+        answer_container = st.container()
+        
+        # Generate content
         response = client.models.generate_content_stream(
-            model="gemini-2.0-flash-thinking-exp-01-21",
+            model=model_name,
             config={'thinking_config': {'include_thoughts': True}},
-            contents=[
-                {'inline_data': {'mime_type': 'image/png', 'data': img_str}},
-                "Resous cette exercice. ça doit être bien présentable et espacé afin d'être facile à lire."
-            ]
+            contents="Give me a practice question I can use for the AP Physics C exam?"
         )
         
+        mode = 'starting'
+        
+        # Process the response
         for chunk in response:
             for part in chunk.candidates[0].content.parts:
                 if part.thought:
-                    thoughts.append(part.text)
+                    if mode != "thinking":
+                        with thinking_container:
+                            st.subheader("Thinking Process")
+                        mode = "thinking"
+                    with thinking_container:
+                        st.write(part.text)
                 else:
-                    answers.append(part.text)
-        
-        return thoughts, answers
-    
-    except Exception as e:
-        st.error(f"Une erreur s'est produite: {str(e)}")
-        return [], []
-
-def main():
-    st.title("Résolution d'Exercices")
-    
-    # File uploader
-    uploaded_file = st.file_uploader("Choisissez une image", type=['png', 'jpg', 'jpeg'])
-    
-    if uploaded_file is not None:
-        # Display the uploaded image
-        image = Image.open(uploaded_file)
-        st.image(image, caption="Image téléchargée", use_column_width=True)
-        
-        # Process button
-        if st.button("Résoudre l'exercice"):
-            with st.spinner("Traitement en cours..."):
-                thoughts, answers = process_image(image)
-                
-                # Display thoughts in expander
-                with st.expander("Voir le raisonnement"):
-                    for i, thought in enumerate(thoughts, 1):
-                        st.markdown(f"**Étape {i}:** {thought}")
-                
-                # Display answer
-                st.markdown("### Solution:")
-                for answer in answers:
-                    st.markdown(answer)
-
-if __name__ == "__main__":
-    main()
+                    if mode != "answering":
+                        with answer_container:
+                            st.subheader("Generated Question")
+                        mode = "answering"
+                    with answer_container:
+                        st.write(part.text)
+else:
+    st.warning("Please enter your Google API key to start generating questions.")