Update app.py

app.py

@@ -6,11 +6,9 @@ import random
 import base64
 import glob
 import math
-#import openai
 import pytz
 import re
 import requests
-import textract
 import time
 import zipfile
 import dotenv
@@ -22,24 +20,20 @@ from datetime import datetime
 from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
 from io import BytesIO
-from openai import ChatCompletion
+
 from PyPDF2 import PdfReader
 from templates import bot_template, css, user_template
 from xml.etree import ElementTree as ET
 from PIL import Image
 from urllib.parse import quote  # Ensure this import is included
 
-#grundle-gpt4o
-
-
-
-import streamlit as st
 import openai
 from openai import OpenAI
-#import os
-#import base64
+
 import cv2
 from moviepy.editor import VideoFileClip
+import textract
+
 
 
 # 1. Configuration
@@ -1583,6 +1577,7 @@ if AddAFileForContext:
 
 
 # Set API key and organization ID from environment variables
+
 openai.api_key = os.getenv('OPENAI_API_KEY')
 openai.organization = os.getenv('OPENAI_ORG_ID')
 client = OpenAI(api_key= os.getenv('OPENAI_API_KEY'), organization=os.getenv('OPENAI_ORG_ID'))
@@ -1603,24 +1598,6 @@ def process_text():
         )
         st.write("Assistant: " + completion.choices[0].message.content)
 
-def process_image_old_05152024(image_input):
-    if image_input:
-        base64_image = base64.b64encode(image_input.read()).decode("utf-8")
-        response = client.chat.completions.create(
-            model=MODEL,
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant that responds in Markdown."},
-                {"role": "user", "content": [
-                    {"type": "text", "text": "Help me understand what is in this picture and list ten facts as markdown outline with appropriate emojis that describes what you see."},
-                    {"type": "image_url", "image_url": {
-                        "url": f"data:image/png;base64,{base64_image}"}
-                    }
-                ]}
-            ],
-            temperature=0.0,
-        )
-        st.markdown(response.choices[0].message.content)
-
 def save_image(image_input, filename):
     # Save the uploaded image file
     with open(filename, "wb") as f:
@@ -1794,9 +1771,6 @@ def process_audio_and_video(video_input):
 
 
 def main():
-
-
-    
     st.markdown("### OpenAI GPT-4o Model")
     st.markdown("#### The Omni Model with Text, Audio, Image, and Video")
     option = st.selectbox("Select an option", ("Text", "Image", "Audio", "Video"))
@@ -1816,28 +1790,64 @@ def main():
     # Image and Video Galleries
     num_columns_video=st.slider(key="num_columns_video", label="Choose Number of Video Columns", min_value=1, max_value=15, value=4)
     display_videos_and_links(num_columns_video)   # Video Jump Grid
-    
     num_columns_images=st.slider(key="num_columns_images", label="Choose Number of Image Columns", min_value=1, max_value=15, value=4)
     display_images_and_wikipedia_summaries(num_columns_images)   # Image Jump Grid
 
 
-
-
-    
-
-if __name__ == "__main__":
-    main()
-
+# Optional UI's
 showExtendedTextInterface=False
 if showExtendedTextInterface:
     display_glossary_grid(roleplaying_glossary)  # Word Glossary Jump Grid - Dynamically calculates columns based on details length to keep topic together
-    
     num_columns_text=st.slider(key="num_columns_text", label="Choose Number of Text Columns", min_value=1, max_value=15, value=4)
     display_buttons_with_scores(num_columns_text)  # Feedback Jump Grid
-        
     st.markdown(personality_factors)
 
 
+def process_text2(MODEL='gpt-4o-2024-05-13', text_input='What is 2+2 and what is an imaginary number'):
+    if text_input:
+        completion = client.chat.completions.create(
+            model=MODEL,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant. Help me with my math homework!"},
+                {"role": "user", "content": f"Hello! Could you solve {text_input}?"}
+            ]
+        )
+        return_text = completion.choices[0].message.content
+        st.write("Assistant: " + return_text)
+        filename = generate_filename(text_input, "md")
+        create_file(filename, text_input, return_text, should_save)
+        return return_text
 
 
+st.title("GPT-4o ChatBot")
+
+client = OpenAI(api_key= os.getenv('OPENAI_API_KEY'), organization=os.getenv('OPENAI_ORG_ID'))
+MODEL = "gpt-4o-2024-05-13"
+if "openai_model" not in st.session_state:
+    st.session_state["openai_model"] = MODEL
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+if st.button("Clear Session"):
+    st.session_state.messages = []
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+if prompt := st.chat_input("What can I help you with?"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    with st.chat_message("assistant"):
+        completion = client.chat.completions.create(
+            model=MODEL,
+            messages=[
+                {"role": m["role"], "content": m["content"]}
+                for m in st.session_state.messages
+            ],
+            stream=True
+        )
+        response = process_text2(text_input=prompt)
+    st.session_state.messages.append({"role": "assistant", "content": response})
 
+if __name__ == "__main__":
+    main()