Update app.py

app.py

@@ -1,168 +1,111 @@
-import os
+import streamlit as st
 import time
-from typing import List, Tuple, Optional
-
-import google.generativeai as genai
-import gradio as gr
+from openai import GPT3, GPT3APIError
+from typing import List
+import io
+import numpy as np
+import matplotlib.pyplot as plt
+import pandas as pd
 from PIL import Image
+import json
+import random
+import base64
+import gr
+
+
+GOOGLE_API_KEY = "your_google_api_key"
+OPENAI_API_KEY = "your_openai_api_key"
+
+st.set_option('deprecation.showfileUploaderEncoding', False)
+
+openai_api = GPT3(engine="text-davinci-002", max_tokens=1024, temperature=0.4, top_p=1, frequency_penalty=0.0, presence_penalty=0.0)
+
+st.title("Gemini")
+
+def main():
+    if GOOGLE_API_KEY is None:
+        st.warning("Please provide your GOOGLE_API_KEY")
+        return
+
+    # Render components
+    global GOOGLE_API_KEY, OPENAI_API_KEY
+    global image_prompt_component, chatbot_component, text_prompt_component, run_button_component
+    global temperature_component, max_output_tokens_component, stop_sequences_component, top_k_component, top_p_component
+
+    google_key_component = st.text_input(
+        label="GOOGLE API KEY",
+        value="",
+        type="password",
+        placeholder="...",
+        help="You have to provide your own GOOGLE_API_KEY for this app to function properly",
+        visible=GOOGLE_API_KEY is None
+    )
+
+    image_prompt_component = st.image_uploader(
+        label="Image",
+        type="pil",
+        accept_jpg=True,
+        accept_png=True,
+        help="Upload an image or drag-and-drop one onto the canvas."
+    )
+
+    chatbot_component = st.empty()
+
+    text_prompt_component = st.text_input(
+        label="",
+        value="",
+        max_chars=500,
+        help="Type your message here and press Enter."
+    )
+
+    run_button_component = st.button(
+        label="Run",
+        help="Generate a response to your input."
+    )
+
+    temperature_component = st.slider(
+        label="Temperature",
+        min_value=0.0,
+        max_value=1.0,
+        value=0.4,
+        step=0.05,
+        help=(
+            "Temperature controls the degree of randomness in token selection. Lower "
+            "temperatures are good for prompts that expect a true or correct response, "
+            "while higher temperatures can lead to more diverse or unexpected results."
+        )
+    )
+
+    max_output_tokens_component = st.slider(
+        label="Token limit",
+        min_value=1,
+        max_value=2048,
+        value=1024,
+        step=1,
+        help=(
+            "Token limit determines the maximum amount of text output from one prompt. A "
+            "token is approximately four characters. The default value is 2048."
+        )
+    )
 
-print("google-generativeai:", genai.__version__)
-
-GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
-
-TITLE = """<h1 align="center" style="color: #4CAF50;">Gemini Playground 😎</h1>"""
-SUBTITLE = """<h2 align="center" style="color: #4CAF50;">Play with Gemini Pro and Gemini Pro Vision API 🖇️</h2>"""
-DUPLICATE = """
-<div style="text-align: center; display: flex; justify-content: center; align-items: center; color: #4CAF50;">
-    <a href="https://huggingface.co/spaces/SkalskiP/ChatGemini?duplicate=true" style="text-decoration: none;">
-        <img src="https://bit.ly/3gLdBN6" alt="Duplicate Space" style="margin-right: 10px;">
-        <span style="font-weight: bold;">Duplicate the Space and run securely with your </span>
-        <a href="https://makersuite.google.com/app/apikey" style="color: #2196F3; font-weight: bold;">GOOGLE API KEY</a>.
-    </a>
-</div>
-"""
-
-
-IMAGE_WIDTH = 512
-
-
-def preprocess_stop_sequences(stop_sequences: str) -> Optional[List[str]]:
-    if not stop_sequences:
-        return None
-    return [sequence.strip() for sequence in stop_sequences.split(",")]
-
-
-def preprocess_image(image: Image.Image) -> Optional[Image.Image]:
-    image_height = int(image.height * IMAGE_WIDTH / image.width)
-    return image.resize((IMAGE_WIDTH, image_height))
-
-
-def user(text_prompt: str, chatbot: List[Tuple[str, str]]):
-    return "", chatbot + [[text_prompt, None]]
-
-
-def bot(
-    google_key: str,
-    image_prompt: Optional[Image.Image],
-    temperature: float,
-    max_output_tokens: int,
-    stop_sequences: str,
-    top_k: int,
-    top_p: float,
-    chatbot: List[Tuple[str, str]]
-):
-    google_key = google_key if google_key else GOOGLE_API_KEY
-    if not google_key:
-        raise ValueError(
-            "GOOGLE_API_KEY is not set. "
-            "Please follow the instructions in the README to set it up.")
-
-    text_prompt = chatbot[-1][0]
-    genai.configure(api_key=google_key)
-    generation_config = genai.types.GenerationConfig(
-        temperature=temperature,
-        max_output_tokens=max_output_tokens,
-        stop_sequences=preprocess_stop_sequences(stop_sequences=stop_sequences),
-        top_k=top_k,
-        top_p=top_p
+    stop_sequences_component = st.text_input(
+        label="Add stop sequence",
+        value="",
+        type="text",
+        placeholder="STOP, END",
+        help=(
+            "A stop sequence is a series of characters (including spaces) that stops "
+            "response generation if the model encounters it. The sequence is not included "
+            "as part of the response. You can add up to five stop sequences."
+        )
     )
 
-    if image_prompt is None:
-        model = genai.GenerativeModel('gemini-pro')
-        response = model.generate_content(
-            text_prompt,
-            stream=True,
-            generation_config=generation_config)
-        response.resolve()
-    else:
-        image_prompt = preprocess_image(image_prompt)
-        model = genai.GenerativeModel('gemini-pro-vision')
-        response = model.generate_content(
-            contents=[text_prompt, image_prompt],
-            stream=True,
-            generation_config=generation_config)
-        response.resolve()
-
-    # streaming effect
-    chatbot[-1][1] = ""
-    for chunk in response:
-        for i in range(0, len(chunk.text), 10):
-            section = chunk.text[i:i + 10]
-            chatbot[-1][1] += section
-            time.sleep(0.01)
-            yield chatbot
-
-
-google_key_component = gr.Textbox(
-    label="GOOGLE API KEY",
-    value="",
-    type="password",
-    placeholder="...",
-    info="You have to provide your own GOOGLE_API_KEY for this app to function properly",
-    visible=GOOGLE_API_KEY is None
-)
-
-image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
-chatbot_component = gr.Chatbot(
-    label='Gemini',
-    bubble_full_width=False,
-    scale=2
-)
-text_prompt_component = gr.Textbox(
-    placeholder="Hi there! Ask me anything and press Enter",
-    label="",
-    style="width: 90%;"
-)
-run_button_component = gr.Button(style="background-color: #4CAF50; color: white; border: none; padding: 10px 20px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 5px;")
-temperature_component = gr.Slider(
-    minimum=0,
-    maximum=1.0,
-    value=0.4,
-    step=0.05,
-    label="Temperature",
-    info=(
-        "Temperature controls the degree of randomness in token selection. Lower "
-        "temperatures are good for prompts that expect a true or correct response, "
-        "while higher temperatures can lead to more diverse or unexpected results."
-    ))
-max_output_tokens_component = gr.Slider(
-    minimum=1,
-    maximum=2048,
-    value=1024,
-    step=1,
-    label="Token limit",
-    info=(
-        "Token limit determines the maximum amount of text output from one prompt. A "
-        "token is approximately four characters. The default value is 2048."
-    ))
-stop_sequences_component = gr.Textbox(
-    label="Add stop sequence",
-    value="",
-    type="text",
-    placeholder="STOP, END",
-    info=(
-        "A stop sequence is a series of characters (including spaces) that stops "
-        "response generation if the model encounters it. The sequence is not included "
-        "as part of the response. You can add up to five stop sequences."
-    ))
-top_k_component = gr.Slider(
-    minimum=1,
-    maximum=40,
-    value=32,
-    step=1,
-    label="Top-K",
-    info=(
-        "Top-k changes how the model selects tokens for output. A top-k of 1 means the "
-        "selected token is the most probable among all tokens in the model’s "
-        "vocabulary (also called greedy decoding), while a top-k of 3 means that the "
-        "next token is selected from among the 3 most probable tokens (using "
-        "temperature)."
-    ))
-top_p_component = gr.Slider(
-    minimum=0,
-    maximum=1,
-    value=1,
-    step=0.01,
-    label="Top-P",
-    info=(
-        "Top
+    top_k_component = st.slider(
+        label="Top-K",
+        min_value=1,
+        max_value=40,
+        value=32,
+        step=1,
+        help=(
+            "Top-k changes how the model selects tokens for output. A top-k of 1 means the "
+            "selected token is the most probable among all tokens in the