File size: 4,975 Bytes
32486dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 |
from pathlib import Path
import streamlit as st
import streamlit.components.v1 as components
from PIL import Image
import base64
import textgrad as tg
from stqdm import stqdm
import json
import os
def read_markdown_file(markdown_file):
return Path(markdown_file).read_text()
def render_svg(svg_filename):
with open(svg_filename,"r") as f:
lines = f.readlines()
svg=''.join(lines)
"""Renders the given svg string."""
b64 = base64.b64encode(svg.encode('utf-8')).decode("utf-8")
html = r'<img src="data:image/svg+xml;base64,%s"/>' % b64
st.write(html, unsafe_allow_html=True)
def load_json(file_path):
with open(file_path, "r") as f:
return json.load(f)
def app():
st.title('TextGrad: Automatic ''Differentiation'' via Text')
st.image('assets/img/textgrad_logo_1x.png', width=200)
st.markdown("### Examples")
examples = {
"Code Editor": load_json("examples/code_editor.json"),
"Example Math Solution": load_json("examples/example_math.json"),
# "🏞️ Image QA": load_json("examples/mathvista.json"),
}
col1, col2 = st.columns([1,1])
with col1:
# Display clickable examples
option = st.selectbox(
"Select an example to use:",
examples.keys()
)
# Initialize session state if not already done
st.session_state.data = examples[option]
# if st.button("Clear example selection"):
# st.session_state.data = ""
# option = "None"
# Initialize session state if not already done
if "selected_option" not in st.session_state:
st.session_state.selected_option = None
st.session_state.object = None
# Check if the selected option has changed
if st.session_state.selected_option != option:
st.session_state.data = examples[option]
st.session_state.selected_option = option
st.session_state.object_initialized = False
# Display the selected option
st.markdown(f"**Example selected:** {option}")
with col2:
valid_api_key = st.session_state.get('valid_api_key', False)
form = st.form(key='api_key')
OPENAI_API_KEY = form.text_input(label='Enter OpenAI API Key. Keys will not be stored', type = 'password')
submit_button = form.form_submit_button(label='Submit')
if submit_button:
os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
try:
engine = tg.get_engine("gpt-4o")
response = engine.generate("hello, what model is this?")
st.session_state.valid_api_key = True
valid_api_key = True
except:
st.error("Please enter a valid OpenAI API key.")
st.stop()
if valid_api_key:
# Initialize the object when the option is selected
if not st.session_state.object_initialized:
if option == "Example Math Solution":
from examples import example_math_scripts
st.session_state.iteration = 0
st.session_state.object = example_math_scripts.MathSolution(data=st.session_state.get("data", ""))
elif option == "Code Editor":
from examples import code_editor_scripts
st.session_state.iteration = 0
st.session_state.object = code_editor_scripts.CodeEditor(data=st.session_state.get("data", ""))
# elif option == "🏞️ Image QA":
# from examples import example_imageQA_scripts
# st.session_state.iteration = 0
# st.session_state.object = example_imageQA_scripts.ImageQA(data=st.session_state.get("data", ""))
st.session_state.object_initialized = True
# Ensure object is loaded
st.session_state.object.load_layout()
# Run TextGrad
# Add custom CSS to style the button
# st.markdown("""
# <style>
# .large-button {
# font-size: 24px;
# padding: 16px 32px;
# background-color: #007bff;
# color: white;
# border: none;
# border-radius: 4px;
# cursor: pointer;
# }
# .large-button:hover {
# background-color: #0056b3;
# }
# </style>
# """, unsafe_allow_html=True)
# # Create a larger button using HTML
# if st.markdown('<button class="large-button">Run 1 iteration of TextGrad</button>', unsafe_allow_html=True):
if st.button("Run 1 iteration of TextGrad"):
st.session_state.object.show_results()
st.markdown("""---""")
st.markdown('### Disclaimer')
st.markdown("Do not get addicted to TextGrad. It is a powerful tool that can be used for good or evil. Use it responsibly.") |