Upload 4 files

README.md

@@ -1,20 +1,23 @@
----
-title: Code Explainer P
-emoji: 🚀
-colorFrom: red
-colorTo: red
-sdk: docker
-app_port: 8501
-tags:
-- streamlit
-pinned: false
-short_description: Streamlit template space
-license: mit
----
+# Code Explainer 🧠
 
-# Welcome to Streamlit!
+This project uses the `bigcode/starcoder` model to explain Python code in plain English.
 
-Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
+## How to Run
 
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+```bash
+pip install -r requirements.txt
+streamlit run app/main.py
+```
+
+## Example
+
+Input:
+```python
+def fibonacci(n):
+    if n <= 1:
+        return n
+    return fibonacci(n-1) + fibonacci(n-2)
+```
+
+Output:
+> This function calculates the nth Fibonacci number using recursion...

prompt_utils.py

@@ -0,0 +1,6 @@
+def build_prompt(code_snippet):
+    return f"""### Explain this Python code step-by-step:
+```python
+{code_snippet}
+```
+Explanation:"""

requirements.txt

@@ -1,3 +1,3 @@
-altair
-pandas
-streamlit
+transformers
+torch
+streamlit

streamlit_app.py

@@ -0,0 +1,21 @@
+import streamlit as st
+from model.model_utils import load_model, generate_explanation
+from app.prompt_utils import build_prompt
+
+st.set_page_config(page_title="Code Explainer", layout="centered")
+
+st.title("🧠 Code Explainer")
+st.write("Paste your Python code below and get a plain English explanation:")
+
+code_input = st.text_area("Paste Python Code", height=200)
+
+if st.button("Explain"):
+    if code_input.strip():
+        with st.spinner("Generating explanation..."):
+            tokenizer, model, device = load_model()
+            prompt = build_prompt(code_input)
+            explanation = generate_explanation(prompt, tokenizer, model, device)
+            st.subheader("📝 Explanation")
+            st.write(explanation.split("Explanation:")[-1].strip())
+    else:
+        st.warning("Please paste some code to explain.")