Update app.py

app.py

@@ -1,17 +1,6 @@
 import streamlit as st
 from transformers import pipeline
 import langchain
-import tensorflow.compat.v1 as tf
-
-# Disable TensorFlow v2 behavior
-tf.disable_v2_behavior()
-
-# Set Streamlit page configuration
-st.set_page_config(
-    page_title="Code Generator",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
 
 def reverse_prompt_engineer(input_code):
     analyzed_code = langchain.analyze(input_code)
@@ -34,6 +23,13 @@ def reverse_prompt_engineer(input_code):
     
     return generated_code
 
+# Set Streamlit page configuration
+st.set_page_config(
+    page_title="Code Generator",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
 # Code Generation Page
 st.title("Code Generator")
 
@@ -50,4 +46,4 @@ if st.button("Generate Code"):
             st.code(generated_code)
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")
-            raise st.StopException(e)
+            raise st.StopException(e)