Update app.py

app.py

@@ -58,6 +58,15 @@ def fix_code_bugs(buggy_code, model="gpt-4o-mini"):
     )
     return response['choices'][0]['message']['content']
 
+from PIL import Image  # Required for local image files
+
+# Streamlit app starts here
+st.set_page_config(page_title="AI Assistance", page_icon=":robot:", layout="wide")
+
+# Display a logo or icon
+image = Image.open("path_to_your_image.png")  # Path to your image file
+st.image(image, width=200)  # You can adjust the width as needed
+
 # Streamlit app starts here
 st.title("AI Assistance")