Update app.py

app.py

@@ -1,9 +1,21 @@
 import streamlit as st
+import google.generativeai as genai
 import requests
 import json
 
-# Gemini API key (replace with your actual API key or credentials)
-gemini_api_key = st.secrets["gemini"]["api_key"]
+MODEL_ID = "gemini-2.0-flash-exp" 
+api_key = os.getenv("GEMINI_API_KEY")
+model_id = MODEL_ID
+genai.configure(api_key=api_key)
+
+if "model" not in st.session_state:
+    st.session_state.model = genai.GenerativeModel(MODEL_ID)
+
+model = st.session_state.model
+chat = model.start_chat()
+
+creative_prompt = ""
+factcheck_prompt = ""
 
 # Page Title
 st.title("LemmaTeks: AI-Powered Text Generator")
@@ -28,16 +40,22 @@ with st.sidebar:
     text_length = st.slider("Text Length (words):", min_value=50, max_value=1000, step=50)
     creativity_level = st.slider("Creativity Level:", min_value=0.0, max_value=1.0, step=0.1)
     
-    # Number of Responses
-    num_responses = st.number_input("Number of Responses:", min_value=1, max_value=5, step=1)
-    
     # Checkboxes for Features
     creative_mode = st.checkbox("Enable Creative Mode")
     fact_checking = st.checkbox("Enable Fact-Checking")
 
+    #modify prompt based on settings
+    if creative_mode:
+        creative_prompt = " Optimize the creativity of your response. "        
+    if fact_checking:
+        factcheck_prompt = "Support your answer with evidences. "
+
 # Text Input Field
 user_prompt = st.text_area("Enter Your Prompt Here:")
 
+# Append the creative and fact checking
+user_prompt = user_prompt + creative_prompt + factcheck_prompt
+
 # Submit Button
 if st.button("Generate"):
     if user_prompt.strip() == "":
@@ -47,32 +65,17 @@ if st.button("Generate"):
         st.write("Generating responses...")
         
         try:
-            # Set up the Gemini API endpoint (this is an example and should be adjusted)
-            api_url = "https://gemini.google.com/app"  # Replace with actual URL
-            headers = {
-                "Authorization": f"Bearer {gemini_api_key}",
-                "Content-Type": "application/json"
-            }
-
-            # Set up the payload for the API request
-            payload = {
-                "model": "gemini",  # Replace with the correct Gemini model name
-                "prompt": f"Format: {output_format}\nTone: {tone_style}\nPrompt: {user_prompt}",
-                "max_tokens": text_length,
-                "temperature": creativity_level,
-                "num_responses": num_responses
-            }
-
-            # Make the API request
-            response = requests.post(api_url, headers=headers, json=payload)
-            response_data = response.json()
-
-            # Display Responses
-            for i, choice in enumerate(response_data["choices"]):
-                st.subheader(f"Response {i + 1}")
-                st.write(choice["text"].strip())
 
-        except requests.exceptions.RequestException as e:
-            st.error(f"An error occurred during the request: {e}")
+            response = model.generate_content(
+                f"Format: {output_format}\nTone: {tone_style}\nPrompt: {user_prompt}",
+                generation_config = genai.GenerationConfig(
+                max_output_tokens=1000,
+                temperature=creativity_level,
+                )                
+            )
+            
+            st.markdown(response.text)
+            
         except Exception as e:
-            st.error(f"An unexpected error occurred: {e}")
+            st.error(f"Exception occured: {e}")
+