Upload folder using huggingface_hub

.gitignore

@@ -0,0 +1,6 @@
+.streamlit/secrets.toml
+src/.streamlit/secrets.toml
+.streamlit/
+src/.streamlit/
+.env
+*.env

Dockerfile

@@ -9,13 +9,35 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Create a non-root user
+RUN useradd -m -u 1000 streamlit
+RUN mkdir -p /home/streamlit/.streamlit
+RUN chown -R streamlit:streamlit /home/streamlit
+
 COPY requirements.txt ./
 COPY src/ ./src/
 
 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)
+ENV AUTH_TOKEN=""
+
+# Set Streamlit configuration via environment variables
+ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
+ENV STREAMLIT_SERVER_PORT=8501
+ENV STREAMLIT_SERVER_ENABLE_CORS=false
+ENV STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION=false
+ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
+ENV STREAMLIT_GLOBAL_DEVELOPMENT_MODE=false
+
 EXPOSE 8501
 
+# Switch to non-root user
+USER streamlit
+
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]

README.md

@@ -1,5 +1,5 @@
 ---
-title: ChainGPT Solidity LLM 2
+title: SolidityLLMDemo
 emoji: 🚀
 colorFrom: red
 colorTo: red

requirements.txt

@@ -1,3 +1,5 @@
-altair
-pandas
-streamlit
+altair
+pandas
+streamlit
+python-dotenv
+requests

src/streamlit_app.py

@@ -1,40 +1,60 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
-
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+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)
+
+if st.button("🚀 Generate Code"):
+    if not prompt.strip():
+        st.warning("Please enter a valid instruction.")
+    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
+                auth_token = os.getenv("AUTH_TOKEN")
+                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: Authentication token not found.")
+            except Exception as e:
+                st.error("❌ An unexpected error occurred. Please try again later.")