Upload folder using huggingface_hub

Dockerfile

@@ -22,8 +22,9 @@ RUN pip3 install -r requirements.txt
 # Change ownership of the app directory
 RUN chown -R streamlit:streamlit /app
 
-# Set environment variable (this will be overridden at runtime)
+# Set environment variables (these will be overridden at runtime)
 ENV AUTH_TOKEN=""
+ENV API_URL=""
 
 # Set Streamlit configuration via environment variables
 ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0

src/streamlit_app.py

@@ -11,18 +11,27 @@ st.set_page_config(page_title="Solidity Code Generator")
 st.title("🛠️ Solidity Code Generator")
 st.write("Enter a coding instruction and get Solidity code")
 
-prompt = st.text_area("💬 Instruction", height=150)
+prompt = st.text_area("💬 Instruction", height=150, max_chars=1000, help="Keep your instruction clear and concise (max 1000 characters)")
 
 if st.button("🚀 Generate Code"):
     if not prompt.strip():
         st.warning("Please enter a valid instruction.")
+    elif len(prompt) > 1000:  # Limit prompt length to prevent model crashes
+        st.warning("⚠️ Your instruction is too long. Please keep it under 1000 characters for better results.")
     else:
         with st.spinner("Generating..."):
             try:
-                api_url = "http://5339246508730216.us-east-1.pai-eas.aliyuncs.com/api/predict/test_llm_service/generate"
-
-                # Get the authorization token from environment variables
+                # Get the API URL and authorization token from environment variables
+                api_url = os.getenv("API_URL")
                 auth_token = os.getenv("AUTH_TOKEN")
+                
+                # Check if required environment variables are set
+                if not api_url:
+                    st.error("❌ Configuration error: API URL not found. Please set the API_URL environment variable.")
+                    st.stop()
+                if not auth_token:
+                    st.error("❌ Configuration error: Authentication token not found. Please set the AUTH_TOKEN environment variable.")
+                    st.stop()
                 headers = {
                     "Authorization": f"Bearer {auth_token}"
                 }
@@ -55,6 +64,6 @@ if st.button("🚀 Generate Code"):
             except requests.exceptions.RequestException as e:
                 st.error("❌ Network error: Unable to connect to the service. Please try again later.")
             except KeyError:
-                st.error("❌ Configuration error: Authentication token not found.")
+                st.error("❌ Configuration error: Required environment variables not found.")
             except Exception as e:
                 st.error("❌ An unexpected error occurred. Please try again later.")