Update app.py

app.py

@@ -1,3 +1,6 @@
+
+
+
 import os
 import streamlit as st
 from groq import Groq
@@ -97,11 +100,43 @@ def store_code_in_faiss(chunks):
     faiss_index.add(np.array(embeddings, dtype=np.float32))
     return faiss_index
 
+# Streamlit App
+def main():
+    # Custom styling for the app using markdown
+    st.markdown("""
+        <style>
+            .title {
+                font-size: 40px;
+                font-weight: bold;
+                color: #FF6347;
+            }
+            .subheader {
+                font-size: 24px;
+                color: #4CAF50;
+            }
+            .generated-code {
+                background-color: #f0f0f0;
+                padding: 10px;
+                border-radius: 10px;
+            }
+            .sidebar .sidebar-content {
+                background-color: #f4f4f4;
+            }
+            .stButton button {
+                background-color: #008CBA;
+                color: white;
+                border-radius: 8px;
+                padding: 10px;
+                font-size: 16px;
+            }
+        </style>
+    """, unsafe_allow_html=True)
+    
     st.title("Microcontroller Code Generator", anchor="title")
 
     # Step 1: Display Custom Image
-    # Use the image URL from Imgur
-    image_url = "https://imgur.com/CBm5exE.png"
+    # Use the image URL from Hugging Face Files Section
+    image_url = "https://imgur.com/a/fq3S6xr.png"
     st.image(image_url, caption="Custom Image", use_container_width=True)
 
     st.write("Select a microcontroller and provide your code requirements to generate code.")
@@ -132,13 +167,6 @@ def store_code_in_faiss(chunks):
             st.subheader("Generated Code", anchor="generated-code")
             st.code(code)
 
-# Add logo to the top bar using the new Imgur image URL
-st.markdown("""
-    <div class="top-bar-logo">
-        <img src="https://i.imgur.com/CBm5exE.png" alt="Logo" width="150">
-    </div>
-""", unsafe_allow_html=True)
-
 # Run the Streamlit App
 if __name__ == "__main__":
     main()