thisisdev commited on
Commit
86a33c4
·
verified ·
1 Parent(s): 139cbb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
app.py CHANGED
@@ -3,17 +3,24 @@ import streamlit as st
3
  from langchain_openai import OpenAI
4
  import getpass
5
  import os
 
 
 
 
6
 
7
  # Creating a function for getting the responses from OpenAI
8
  def get_response(question):
9
- os.environ["OPENAI_API_KEY"] = keyfile.OPENAI_API_KEY
10
- llm = OpenAI()
11
- answer = llm.invoke(question)
12
- return answer
 
 
 
13
 
14
  # Using Streamlit for generation of page
15
- st.set_page_config(page_title = "ASK GPT", page_icon = ":robot:")
16
- st.header("ASK GPT Application")
17
 
18
  # Create a function for taking user input
19
  def get_user_input():
 
3
  from langchain_openai import OpenAI
4
  import getpass
5
  import os
6
+ import google.generativeai as genai
7
+
8
+
9
+ genai.configure(api_key=keyfile.GEMINI_API_KEY)
10
 
11
  # Creating a function for getting the responses from OpenAI
12
  def get_response(question):
13
+ # os.environ["OPENAI_API_KEY"] = keyfile.OPENAI_API_KEY
14
+ # llm = OpenAI()
15
+ # answer = llm.invoke(question)
16
+ # return answer
17
+ model = genai.GenerativeModel("gemini-1.5-flash")
18
+ response = model.generate_content("Write a story about a magic backpack.")
19
+ return response.text
20
 
21
  # Using Streamlit for generation of page
22
+ st.set_page_config(page_title = "ASK GPT / GEMINI", page_icon = ":robot:")
23
+ st.header("ASK GPT / GEMINI Application")
24
 
25
  # Create a function for taking user input
26
  def get_user_input():