Update app.py

app.py

@@ -1,219 +1,216 @@
-import os
-import asyncio
-# Init with fake key
-if 'OPENAI_API_KEY' not in os.environ:
-    os.environ['OPENAI_API_KEY'] = 'none'
-if os.name == 'nt':
-    asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
-    
-import openai
-import pandas as pd
-import streamlit as st
-from IPython.core.display import HTML
-from PIL import Image
-from  agent import TeLLAgent, make_tools
-from streamlit_callback_handler import \
-    StreamlitCallbackHandlerChem
-import base64 
-import pandas as pd
-from dotenv import load_dotenv
-from langchain_openai import ChatOpenAI , OpenAI
-import base64
-from io import BytesIO
-from PIL import Image
-import tempfile
- 
-
-def convert_to_base64(pil_image):
-    buffered = BytesIO()
-    pil_image.save(buffered, format="PNG")
-    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
-    return img_str
-
-def oai_key_isvalid(api_key):
-    """Check if a given OpenAI key is valid"""
-    try:
-        llm = ChatOpenAI(openai_api_key = api_key, base_url="https://www.dmxapi.com/v1/")
-        out = llm.invoke("This is a test")
-        return True
-    except:
-        return False
-    
-load_dotenv()
-ss = st.session_state
-ss.prompt = None
- 
-# Set width of sidebar
-st.markdown(
-    """
-    <style>
-    [data-testid="stSidebar"][aria-expanded="true"]{
-        min-width: 450px;
-        max-width: 450px;
-    }
-    """,
-    unsafe_allow_html=True,
-)
-
-
-def instantiate_agent(model,file_path = '...',
-        image_path ='...'):
-    ss.agent = TeLLAgent(
-        model=model,
-        tools_model=model,
-        temp=0.1,
-        openai_api_key=ss.get('api_key')    ,        file_path = file_path,
-                image_path =image_path
-      
-    )
-    return ss.agent
-
-instantiate_agent('gpt-4o-2024-11-20')
-tools = ss.agent.agent_executor.tools
-
-tool_list = pd.Series(
-    {f"✅ {t.name}":t.description for t in tools}
-).reset_index()
-tool_list.columns = ['Tool', 'Description']
-
-def on_api_key_change():
-    api_key = ss.get('api_key') or os.getenv('OPENAI_API_KEY')
-     
-    # Check if key is valid
-    if not oai_key_isvalid(api_key):
-        st.write("Please input a valid OpenAI API key.")
-
-def run_prompt(prompt, file_path = '...', image_path = '...'):
-    agent = instantiate_agent(ss.get('model_select'),file_path = file_path, image_path =image_path)
-    st.chat_message("user").write(prompt)
-    with st.chat_message("assistant") :
-        try:
-           
-           response = agent.run(prompt)
-           if ss.get('file_type') == 'CSV (.csv)':
-               try: 
-                   fx = pd.DataFrame(list(response))
-                   st.markdown(":red[Prediction finished! ]")	
-                   st.download_button( "⬇️Download  the predicted files as .csv", fx.to_csv(), "predict results.csv", use_container_width=True)
-               except:
-                   st.write(response)
-           else:
-                st.write(response)
-        except openai.AuthenticationError:
-            st.write("Please input a valid OpenAI API key")
-        except openai.APIError:
-            # Handle specific API errors here
-            print("OpenAI API error, please try again!")
-    
-
-pre_prompts = [
-    'Who are you?',
-    ('The history and development of Y6'
-       
-    ),
-    (
-        'Predict the LogP of Y6'
-    ),
-    'Generate a donor material with PCE = 10'
-]
-
-# sidebar
-with st.sidebar:
-
-    st.header("🤖 :blue[TeLLAgent] ")
-    # Input OpenAI api key
-    st.text_input(
-        'Input your OpenAI API key.',
-        placeholder = 'Input your OpenAI API key.',
-        type='password',
-        key='api_key',
-        on_change=on_api_key_change,
-        label_visibility="collapsed"
-    )
- 
-    # Input model to use
-    st.selectbox(
-        'Select model to use',
-        ['gpt-4o-2024-11-20', 'deepseek-v3', 'gpt-4o-mini'],
-        key='model_select',
-    )
-
-    # Display prompt examples
-    st.markdown('# What can I ask?')
-    cols = st.columns(2)
-    with cols[0]:
-        st.button(
-           r'👑  Who     are    you  ?  🧨               ',
-            on_click=lambda: run_prompt(pre_prompts[0]),
-        )
-        st.button(
-            r'📚 The history and development of Y6     ',
-            on_click=lambda: run_prompt(pre_prompts[1]),
-        )
-    with cols[1]:
-        st.button(
-            r"🎄Predict the LogP of Y6  ",
-            on_click=lambda: run_prompt(pre_prompts[2]),
-        )
-        st.button(
-            r'💎 Generate a donor material with PCE = 10',
-            on_click=lambda: run_prompt(pre_prompts[3]),
-        )
-
-    st.selectbox(
-            'Select the file type ',
-            ['None', 'CSV (.csv)', 'Figure (.jpg, .png, .jpeg)', 'PDF (.pdf)'],
-            key='file_type',
-        )
-    uploaded_file = None
-    if ss.get('file_type') == 'Figure (.jpg, .png, .jpeg)':
-        uploaded_file = st.file_uploader("Choose a Figure", type = ["jpg", "jpeg", "png"])
-    if ss.get('file_type') == 'PDF (.pdf)':
-        uploaded_file = st.file_uploader("Choose a PDF file")
-    if ss.get('file_type') == 'CSV (.csv)':
-        uploaded_file = st.file_uploader("Choose a csv file", type = 'csv')   
-        
-    # Display available tools
-    st.markdown(f"# {len(tool_list)} available tools")
-    st.dataframe(
-        tool_list,
-        use_container_width=True,
-        hide_index=True,
-        height=200
-    )
-
-# Execute agent on user input
-if prompt := st.chat_input("Say something and/or attach files"):
-
-    if uploaded_file is not None:
-        if ss.get('file_type') == 'CSV (.csv)':
-            with tempfile.NamedTemporaryFile( dir = 'j:/', suffix ='.csv' ,delete=False) as f:
-                 f.write(uploaded_file.read())
-                 run_prompt(prompt + str(' ') +  str(f.name), file_path =  f.name) 
-                 f.close()
-      
-        if ss.get('file_type') == 'Figure (.jpg, .png, .jpeg)':
-    
-            st.image(uploaded_file, width = 500)
-            with tempfile.NamedTemporaryFile(dir = 'j:/',delete=False, suffix=".png") as temp:
-    
-                   mg_str = base64.b64encode(uploaded_file.getvalue()).decode("utf-8")
-                   temp.write(base64.b64decode(mg_str))
-             
-            run_prompt(prompt+ str(' ') + str(temp.name), image_path = temp.name )
-   
-        if ss.get('file_type') == 'PDF (.pdf)':
-                with tempfile.NamedTemporaryFile( dir = 'j:/', suffix ='.pdf' ,delete=False) as f:
-                     f.write(uploaded_file.read())
-                     run_prompt(prompt, file_path =  f.name) 
-                     f.close()
-      
-                    # with open("input.png","wb") as af:
-                    #              mg_str = base64.b64encode(files.getvalue()).decode("utf-8")
-                    #              af.write(base64.b64decode(mg_str))        
-                     
-                    # run_prompt(prompt.text+str(f.name), image_path =f.name )
-                # except:
-                #     st.markdown("Please input correct files or query ")
-    else:
-          run_prompt(prompt)
+import os
+import asyncio
+# Init with fake key
+if 'OPENAI_API_KEY' not in os.environ:
+    os.environ['OPENAI_API_KEY'] = 'none'
+if os.name == 'nt':
+    asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
+
+import pandas as pd
+import streamlit as st
+from IPython.core.display import HTML
+from PIL import Image
+from  agent import TeLLAgent, make_tools
+from streamlit_callback_handler import \
+    StreamlitCallbackHandlerChem
+import base64 
+import pandas as pd
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI , OpenAI
+import base64
+from io import BytesIO
+from PIL import Image
+import tempfile
+ 
+
+def convert_to_base64(pil_image):
+    buffered = BytesIO()
+    pil_image.save(buffered, format="PNG")
+    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return img_str
+
+def oai_key_isvalid(api_key):
+    """Check if a given OpenAI key is valid"""
+    try:
+        llm = ChatOpenAI(openai_api_key = api_key, base_url="https://www.dmxapi.com/v1/")
+        out = llm.invoke("This is a test")
+        return True
+    except:
+        return False
+    
+load_dotenv()
+ss = st.session_state
+ss.prompt = None
+ 
+# Set width of sidebar
+st.markdown(
+    """
+    <style>
+    [data-testid="stSidebar"][aria-expanded="true"]{
+        min-width: 450px;
+        max-width: 450px;
+    }
+    """,
+    unsafe_allow_html=True,
+)
+
+
+def instantiate_agent(model,file_path = '...',
+        image_path ='...'):
+    ss.agent = TeLLAgent(
+        model=model,
+        tools_model=model,
+        temp=0.1,
+        openai_api_key=ss.get('api_key')    ,        file_path = file_path,
+                image_path =image_path
+      
+    )
+    return ss.agent
+
+instantiate_agent('gpt-4o-2024-11-20')
+tools = ss.agent.agent_executor.tools
+
+tool_list = pd.Series(
+    {f"✅ {t.name}":t.description for t in tools}
+).reset_index()
+tool_list.columns = ['Tool', 'Description']
+
+def on_api_key_change():
+    api_key = ss.get('api_key') or os.getenv('OPENAI_API_KEY')
+     
+    # Check if key is valid
+    if not oai_key_isvalid(api_key):
+        st.write("Please input a valid OpenAI API key.")
+
+def run_prompt(prompt, file_path = '...', image_path = '...'):
+    agent = instantiate_agent(ss.get('model_select'),file_path = file_path, image_path =image_path)
+    st.chat_message("user").write(prompt)
+    with st.chat_message("assistant") :
+        try:
+           
+           response = agent.run(prompt)
+           if ss.get('file_type') == 'CSV (.csv)':
+               try: 
+                   fx = pd.DataFrame(list(response))
+                   st.markdown(":red[Prediction finished! ]")	
+                   st.download_button( "⬇️Download  the predicted files as .csv", fx.to_csv(), "predict results.csv", use_container_width=True)
+               except:
+                   st.write(response)
+           else:
+                st.write(response)
+        except:
+            st.write("Please input a valid OpenAI API key")
+
+    
+
+pre_prompts = [
+    'Who are you?',
+    ('The history and development of Y6'
+       
+    ),
+    (
+        'Predict the LogP of Y6'
+    ),
+    'Generate a donor material with PCE = 10'
+]
+
+# sidebar
+with st.sidebar:
+
+    st.header("🤖 :blue[TeLLAgent] ")
+    # Input OpenAI api key
+    st.text_input(
+        'Input your OpenAI API key.',
+        placeholder = 'Input your OpenAI API key.',
+        type='password',
+        key='api_key',
+        on_change=on_api_key_change,
+        label_visibility="collapsed"
+    )
+ 
+    # Input model to use
+    st.selectbox(
+        'Select model to use',
+        ['gpt-4o-2024-11-20', 'deepseek-v3', 'gpt-4o-mini'],
+        key='model_select',
+    )
+
+    # Display prompt examples
+    st.markdown('# What can I ask?')
+    cols = st.columns(2)
+    with cols[0]:
+        st.button(
+           r'👑  Who     are    you  ?  🧨               ',
+            on_click=lambda: run_prompt(pre_prompts[0]),
+        )
+        st.button(
+            r'📚 The history and development of Y6     ',
+            on_click=lambda: run_prompt(pre_prompts[1]),
+        )
+    with cols[1]:
+        st.button(
+            r"🎄Predict the LogP of Y6  ",
+            on_click=lambda: run_prompt(pre_prompts[2]),
+        )
+        st.button(
+            r'💎 Generate a donor material with PCE = 10',
+            on_click=lambda: run_prompt(pre_prompts[3]),
+        )
+
+    st.selectbox(
+            'Select the file type ',
+            ['None', 'CSV (.csv)', 'Figure (.jpg, .png, .jpeg)', 'PDF (.pdf)'],
+            key='file_type',
+        )
+    uploaded_file = None
+    if ss.get('file_type') == 'Figure (.jpg, .png, .jpeg)':
+        uploaded_file = st.file_uploader("Choose a Figure", type = ["jpg", "jpeg", "png"])
+    if ss.get('file_type') == 'PDF (.pdf)':
+        uploaded_file = st.file_uploader("Choose a PDF file")
+    if ss.get('file_type') == 'CSV (.csv)':
+        uploaded_file = st.file_uploader("Choose a csv file", type = 'csv')   
+        
+    # Display available tools
+    st.markdown(f"# {len(tool_list)} available tools")
+    st.dataframe(
+        tool_list,
+        use_container_width=True,
+        hide_index=True,
+        height=200
+    )
+
+# Execute agent on user input
+if prompt := st.chat_input("Say something and/or attach files"):
+
+    if uploaded_file is not None:
+        if ss.get('file_type') == 'CSV (.csv)':
+            with tempfile.NamedTemporaryFile( dir = 'j:/', suffix ='.csv' ,delete=False) as f:
+                 f.write(uploaded_file.read())
+                 run_prompt(prompt + str(' ') +  str(f.name), file_path =  f.name) 
+                 f.close()
+      
+        if ss.get('file_type') == 'Figure (.jpg, .png, .jpeg)':
+    
+            st.image(uploaded_file, width = 500)
+            with tempfile.NamedTemporaryFile(dir = 'j:/',delete=False, suffix=".png") as temp:
+    
+                   mg_str = base64.b64encode(uploaded_file.getvalue()).decode("utf-8")
+                   temp.write(base64.b64decode(mg_str))
+             
+            run_prompt(prompt+ str(' ') + str(temp.name), image_path = temp.name )
+   
+        if ss.get('file_type') == 'PDF (.pdf)':
+                with tempfile.NamedTemporaryFile( dir = 'j:/', suffix ='.pdf' ,delete=False) as f:
+                     f.write(uploaded_file.read())
+                     run_prompt(prompt, file_path =  f.name) 
+                     f.close()
+      
+                    # with open("input.png","wb") as af:
+                    #              mg_str = base64.b64encode(files.getvalue()).decode("utf-8")
+                    #              af.write(base64.b64decode(mg_str))        
+                     
+                    # run_prompt(prompt.text+str(f.name), image_path =f.name )
+                # except:
+                #     st.markdown("Please input correct files or query ")
+    else:
+          run_prompt(prompt)