Update app.py

app.py

@@ -130,6 +130,49 @@ def reset_conversation():
     st.session_state.conversation = []
     st.session_state.messages = []
 
+# Function to generate filenames
+def generate_filename(prompt, file_type):
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+    replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
+    safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
+    return f"{safe_date_time}_{safe_prompt}.{file_type}"
+
+# Function to create files
+def create_file(filename, prompt, response, user_name, timestamp):
+    with open(filename, "w", encoding="utf-8") as f:
+        f.write(f"User: {user_name}\nTimestamp: {timestamp}\n\nPrompt:\n{prompt}\n\nResponse:\n{response}")
+
+# Function to extract video frames
+def extract_video_frames(video_path, seconds_per_frame=2):
+    base64Frames = []
+    video = cv2.VideoCapture(video_path)
+    total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
+    fps = video.get(cv2.CAP_PROP_FPS)
+    frames_to_skip = int(fps * seconds_per_frame)
+    curr_frame = 0
+    while curr_frame < total_frames - 1:
+        video.set(cv2.CAP_PROP_POS_FRAMES, curr_frame)
+        success, frame = video.read()
+        if not success:
+            break
+        _, buffer = cv2.imencode(".jpg", frame)
+        base64Frames.append(base64.b64encode(buffer).decode("utf-8"))
+        curr_frame += frames_to_skip
+    video.release()
+    return base64Frames, None
+
+# Function to process audio for video
+def process_audio_for_video(video_input):
+    try:
+        transcription = client.audio.transcriptions.create(
+            model="whisper-1",
+            file=video_input,
+        )
+        return transcription.text
+    except:
+        return ''
+
 # Function to process text with selected model
 def process_text(user_name, text_input, selected_model, temp_values):
     timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
@@ -283,7 +326,7 @@ def main_column(column_name):
             process_video(st.session_state.current_user['name'], video_input, text_input)
 
 # Main Streamlit app
-st.title("Multiuser Chat with Llama 3.1 and GPT-4o")
+st.title("Personalized Real-Time Chat")
 
 # Sidebar
 with st.sidebar:
@@ -319,50 +362,7 @@ with col1:
 with col2:
     main_column("Column 2")
 
-# Function to generate filenames
-def generate_filename(prompt, file_type):
-    central = pytz.timezone('US/Central')
-    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
-    replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
-    safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
-    return f"{safe_date_time}_{safe_prompt}.{file_type}"
-
-# Function to create files
-def create_file(filename, prompt, response, user_name, timestamp):
-    with open(filename, "w", encoding="utf-8") as f:
-        f.write(f"User: {user_name}\nTimestamp: {timestamp}\n\nPrompt:\n{prompt}\n\nResponse:\n{response}")
-
-# Function to extract video frames
-def extract_video_frames(video_path, seconds_per_frame=2):
-    base64Frames = []
-    video = cv2.VideoCapture(video_path)
-    total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
-    fps = video.get(cv2.CAP_PROP_FPS)
-    frames_to_skip = int(fps * seconds_per_frame)
-    curr_frame = 0
-    while curr_frame < total_frames - 1:
-        video.set(cv2.CAP_PROP_POS_FRAMES, curr_frame)
-        success, frame = video.read()
-        if not success:
-            break
-        _, buffer = cv2.imencode(".jpg", frame)
-        base64Frames.append(base64.b64encode(buffer).decode("utf-8"))
-        curr_frame += frames_to_skip
-    video.release()
-    return base64Frames, None
-
-# Function to process audio for video
-def process_audio_for_video(video_input):
-    try:
-        transcription = client.audio.transcriptions.create(
-            model="whisper-1",
-            file=video_input,
-        )
-        return transcription.text
-    except:
-        return ''
-
 # Run the Streamlit app
 if __name__ == "__main__":
-    st.markdown("*Generated content may be inaccurate or false.*")
-    st.markdown("\n...")            
+    st.markdown("*by Aaron Wacker*")
+    st.markdown("\n[Aaron Wacker](https://huggingface.co/spaces/awacke1/).")