Ci-Dave commited on
Commit
75cd008
·
1 Parent(s): f4067a3

change api code from openai to gemini

Browse files
Files changed (1) hide show
  1. app.py +29 -21
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import streamlit as st
2
- import openai
 
3
 
4
- # OpenAI API key
5
- openai.api_key = st.secrets["openai"]["api_key"]
6
 
7
  # Page Title
8
  st.title("LemmaTeks: AI-Powered Text Generator")
@@ -46,25 +47,32 @@ if st.button("Generate"):
46
  st.write("Generating responses...")
47
 
48
  try:
49
- # OpenAI API call
50
- response = openai.Completion.create(
51
- engine="text-davinci-003", # GPT-3.5 engine
52
- prompt=f"Format: {output_format}\nTone: {tone_style}\nPrompt: {user_prompt}",
53
- max_tokens=text_length, # Approximate word count
54
- temperature=creativity_level, # Creativity level
55
- n=num_responses # Number of responses
56
- )
57
-
 
 
 
 
 
 
 
 
 
 
 
58
  # Display Responses
59
- for i, choice in enumerate(response.choices):
60
  st.subheader(f"Response {i + 1}")
61
- st.write(choice.text.strip())
62
-
63
- except openai.error.AuthenticationError as e:
64
- st.error(f"Authentication failed. Please check your API key: {e}")
65
- except openai.error.RateLimitError as e:
66
- st.error(f"Rate limit exceeded. Please try again later: {e}")
67
- except openai.OpenAIError as e: # This is the correct way for general OpenAI errors
68
- st.error(f"OpenAI API error: {e}")
69
  except Exception as e:
70
  st.error(f"An unexpected error occurred: {e}")
 
1
  import streamlit as st
2
+ import requests
3
+ import json
4
 
5
+ # Gemini API key (replace with your actual API key or credentials)
6
+ gemini_api_key = st.secrets["gemini"]["api_key"]
7
 
8
  # Page Title
9
  st.title("LemmaTeks: AI-Powered Text Generator")
 
47
  st.write("Generating responses...")
48
 
49
  try:
50
+ # Set up the Gemini API endpoint (this is an example and should be adjusted)
51
+ api_url = "https://gemini.google.com/app" # Replace with actual URL
52
+ headers = {
53
+ "Authorization": f"Bearer {gemini_api_key}",
54
+ "Content-Type": "application/json"
55
+ }
56
+
57
+ # Set up the payload for the API request
58
+ payload = {
59
+ "model": "gemini", # Replace with the correct Gemini model name
60
+ "prompt": f"Format: {output_format}\nTone: {tone_style}\nPrompt: {user_prompt}",
61
+ "max_tokens": text_length,
62
+ "temperature": creativity_level,
63
+ "num_responses": num_responses
64
+ }
65
+
66
+ # Make the API request
67
+ response = requests.post(api_url, headers=headers, json=payload)
68
+ response_data = response.json()
69
+
70
  # Display Responses
71
+ for i, choice in enumerate(response_data["choices"]):
72
  st.subheader(f"Response {i + 1}")
73
+ st.write(choice["text"].strip())
74
+
75
+ except requests.exceptions.RequestException as e:
76
+ st.error(f"An error occurred during the request: {e}")
 
 
 
 
77
  except Exception as e:
78
  st.error(f"An unexpected error occurred: {e}")