Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import requests | |
| import os | |
| from dotenv import load_dotenv | |
| # Load environment variables | |
| load_dotenv() | |
| 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, 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: | |
| # 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}" | |
| } | |
| response = requests.post(api_url, json={"prompt": prompt}, stream=True, headers=headers) | |
| response.raise_for_status() | |
| # Get only the code from the JSON response | |
| result = response.json() | |
| solidity_code = result.get("code", "") | |
| if solidity_code: | |
| st.code(solidity_code, language="solidity") | |
| else: | |
| st.warning("Received an empty response from the Deployed Model.") | |
| except requests.exceptions.Timeout: | |
| st.error("❌ Request timed out. The service is taking too long to respond. Please try again.") | |
| except requests.exceptions.ConnectionError: | |
| st.error("❌ Network error: Unable to connect to the service. Please check your internet connection.") | |
| except requests.exceptions.HTTPError as e: | |
| if e.response.status_code == 503: | |
| st.error("❌ Service temporarily unavailable. The AI service is currently down. Please try again later.") | |
| elif e.response.status_code == 401: | |
| st.error("❌ Authentication failed. Please check your API token.") | |
| elif e.response.status_code == 403: | |
| st.error("❌ Access denied. Please check your API permissions.") | |
| else: | |
| st.error(f"❌ HTTP Error {e.response.status_code}: {e.response.text}") | |
| 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: Required environment variables not found.") | |
| except Exception as e: | |
| st.error("❌ An unexpected error occurred. Please try again later.") |