Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,6 @@ EXAMPLE_QUESTIONS = [
|
|
| 20 |
"How can I make this code more readable?"
|
| 21 |
]
|
| 22 |
|
| 23 |
-
# --- API CALLS ---
|
| 24 |
def call_groq_api(prompt, model="llama3-70b-8192"):
|
| 25 |
headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
|
| 26 |
data = {"model": model, "messages": [{"role": "user", "content": prompt}]}
|
|
@@ -46,7 +45,6 @@ def call_blackbox_agent(messages):
|
|
| 46 |
else:
|
| 47 |
return call_groq_api(messages[-1]["content"])
|
| 48 |
|
| 49 |
-
# --- UTILS ---
|
| 50 |
def code_matches_language(code, language):
|
| 51 |
if language.lower() in code.lower():
|
| 52 |
return True
|
|
@@ -77,72 +75,12 @@ def is_coding_question(question):
|
|
| 77 |
except Exception:
|
| 78 |
return False
|
| 79 |
|
| 80 |
-
# --- STREAMLIT APP ---
|
| 81 |
st.set_page_config(page_title="AI Workflow App", layout="wide")
|
| 82 |
st.title("AI Assistant with Workflow (Streamlit Edition)")
|
| 83 |
|
| 84 |
-
# Navigation
|
| 85 |
page = st.sidebar.radio("Navigate", ["Home", "AI Workflow", "Semantic Search"])
|
| 86 |
|
| 87 |
-
if page == "
|
| 88 |
-
st.header("Welcome to the AI Assistant!")
|
| 89 |
-
st.markdown("""
|
| 90 |
-
- **Full AI Workflow:** Complete code analysis pipeline with explanation, refactoring, review, and testing (powered by Groq/Blackbox)
|
| 91 |
-
- **Semantic Search:** Ask natural language questions about your code and get intelligent answers
|
| 92 |
-
""")
|
| 93 |
-
st.info("Select a feature from the sidebar to get started.")
|
| 94 |
-
|
| 95 |
-
elif page == "AI Workflow":
|
| 96 |
-
st.header("Full AI Workflow")
|
| 97 |
-
code_input = st.text_area("Paste your code here", height=200)
|
| 98 |
-
uploaded_file = st.file_uploader("Or upload a code file", type=["py", "js", "ts", "java", "cpp", "cs"])
|
| 99 |
-
if uploaded_file:
|
| 100 |
-
code_input = uploaded_file.read().decode("utf-8")
|
| 101 |
-
st.text_area("File content", code_input, height=200, key="file_content")
|
| 102 |
-
col1, col2, col3, col4 = st.columns(4)
|
| 103 |
-
with col1:
|
| 104 |
-
programming_language = st.selectbox("Programming Language", PROGRAMMING_LANGUAGES)
|
| 105 |
-
with col2:
|
| 106 |
-
skill_level = st.selectbox("Skill Level", SKILL_LEVELS)
|
| 107 |
-
with col3:
|
| 108 |
-
user_role = st.selectbox("Your Role", USER_ROLES)
|
| 109 |
-
with col4:
|
| 110 |
-
explanation_language = st.selectbox("Explanation Language", EXPLANATION_LANGUAGES)
|
| 111 |
-
if code_input:
|
| 112 |
-
st.caption(f"Complexity: {calculate_code_complexity(code_input)}")
|
| 113 |
-
if st.button("Run Workflow", type="primary"):
|
| 114 |
-
if not code_input.strip():
|
| 115 |
-
st.error("Please paste or upload your code.")
|
| 116 |
-
elif not code_matches_language(code_input, programming_language):
|
| 117 |
-
st.error(f"Language mismatch. Please check your code and language selection.")
|
| 118 |
-
else:
|
| 119 |
-
with st.spinner("Running AI Workflow..."):
|
| 120 |
-
steps = [
|
| 121 |
-
("Explain", call_groq_api(f"Explain this {programming_language} code for a {skill_level} {user_role} in {explanation_language}:\n{code_input}")),
|
| 122 |
-
("Refactor", call_blackbox_agent([
|
| 123 |
-
{"role": "system", "content": "You are a helpful coding assistant."},
|
| 124 |
-
{"role": "user", "content": f"Refactor this {programming_language} code: {code_input}"}
|
| 125 |
-
])),
|
| 126 |
-
("Review", call_groq_api(f"Review this {programming_language} code for errors and improvements: {code_input}")),
|
| 127 |
-
("ErrorDetection", call_groq_api(f"Find bugs in this {programming_language} code: {code_input}")),
|
| 128 |
-
("TestGeneration", call_groq_api(f"Generate tests for this {programming_language} code: {code_input}")),
|
| 129 |
-
]
|
| 130 |
-
timeline = []
|
| 131 |
-
for step, output in steps:
|
| 132 |
-
timeline.append({"step": step, "output": output})
|
| 133 |
-
st.success("Workflow complete!")
|
| 134 |
-
for t in timeline:
|
| 135 |
-
st.subheader(t["step"])
|
| 136 |
-
st.write(t["output"])
|
| 137 |
-
st.subheader("Code Diff (Original vs Refactored)")
|
| 138 |
-
refactored_code = steps[1][1]
|
| 139 |
-
st.code(get_inline_diff(code_input, refactored_code), language=programming_language.lower())
|
| 140 |
-
report = f"AI Workflow Report\nGenerated on: {datetime.datetime.now()}\nLanguage: {programming_language}\nSkill Level: {skill_level}\nRole: {user_role}\n\n"
|
| 141 |
-
for t in timeline:
|
| 142 |
-
report += f"## {t['step']}\n{t['output']}\n\n---\n\n"
|
| 143 |
-
st.download_button("Download Report", report, file_name="ai_workflow_report.txt")
|
| 144 |
-
|
| 145 |
-
elif page == "Semantic Search":
|
| 146 |
st.header("Semantic Search")
|
| 147 |
code_input = st.text_area("Paste your code here", height=200, key="sem_code")
|
| 148 |
uploaded_file = st.file_uploader("Or upload a code file", type=["py", "js", "ts", "java", "cpp", "cs"], key="sem_file")
|
|
@@ -162,6 +100,47 @@ elif page == "Semantic Search":
|
|
| 162 |
st.caption("Example questions:")
|
| 163 |
st.write(", ".join(EXAMPLE_QUESTIONS))
|
| 164 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
question = st.text_input("Ask a question about your code", key="sem_question")
|
| 166 |
|
| 167 |
if st.button("Run Semantic Search"):
|
|
|
|
| 20 |
"How can I make this code more readable?"
|
| 21 |
]
|
| 22 |
|
|
|
|
| 23 |
def call_groq_api(prompt, model="llama3-70b-8192"):
|
| 24 |
headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
|
| 25 |
data = {"model": model, "messages": [{"role": "user", "content": prompt}]}
|
|
|
|
| 45 |
else:
|
| 46 |
return call_groq_api(messages[-1]["content"])
|
| 47 |
|
|
|
|
| 48 |
def code_matches_language(code, language):
|
| 49 |
if language.lower() in code.lower():
|
| 50 |
return True
|
|
|
|
| 75 |
except Exception:
|
| 76 |
return False
|
| 77 |
|
|
|
|
| 78 |
st.set_page_config(page_title="AI Workflow App", layout="wide")
|
| 79 |
st.title("AI Assistant with Workflow (Streamlit Edition)")
|
| 80 |
|
|
|
|
| 81 |
page = st.sidebar.radio("Navigate", ["Home", "AI Workflow", "Semantic Search"])
|
| 82 |
|
| 83 |
+
if page == "Semantic Search":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 84 |
st.header("Semantic Search")
|
| 85 |
code_input = st.text_area("Paste your code here", height=200, key="sem_code")
|
| 86 |
uploaded_file = st.file_uploader("Or upload a code file", type=["py", "js", "ts", "java", "cpp", "cs"], key="sem_file")
|
|
|
|
| 100 |
st.caption("Example questions:")
|
| 101 |
st.write(", ".join(EXAMPLE_QUESTIONS))
|
| 102 |
|
| 103 |
+
# --- Voice input button (for local use only) ---
|
| 104 |
+
st.markdown("""
|
| 105 |
+
<button id="voice-btn" style="font-size:22px; margin-bottom:8px;">🎤 Speak your question</button>
|
| 106 |
+
<span id="voice-status" style="margin-left:8px;"></span>
|
| 107 |
+
<script>
|
| 108 |
+
const btn = document.getElementById('voice-btn');
|
| 109 |
+
const status = document.getElementById('voice-status');
|
| 110 |
+
let recognition;
|
| 111 |
+
if ('webkitSpeechRecognition' in window) {
|
| 112 |
+
recognition = new webkitSpeechRecognition();
|
| 113 |
+
recognition.lang = 'en-US';
|
| 114 |
+
recognition.continuous = false;
|
| 115 |
+
recognition.interimResults = false;
|
| 116 |
+
btn.onclick = function() {
|
| 117 |
+
recognition.start();
|
| 118 |
+
status.textContent = 'Listening...';
|
| 119 |
+
};
|
| 120 |
+
recognition.onresult = function(event) {
|
| 121 |
+
const transcript = event.results[0][0].transcript;
|
| 122 |
+
// Find the Streamlit input and set its value
|
| 123 |
+
const streamlitInput = window.parent.document.querySelector('input[data-testid="stTextInput"]');
|
| 124 |
+
if (streamlitInput) {
|
| 125 |
+
streamlitInput.value = transcript;
|
| 126 |
+
streamlitInput.dispatchEvent(new Event('input', { bubbles: true }));
|
| 127 |
+
}
|
| 128 |
+
status.textContent = 'Heard: ' + transcript;
|
| 129 |
+
};
|
| 130 |
+
recognition.onerror = function() {
|
| 131 |
+
status.textContent = 'Voice error';
|
| 132 |
+
};
|
| 133 |
+
recognition.onend = function() {
|
| 134 |
+
if (status.textContent === 'Listening...') status.textContent = '';
|
| 135 |
+
};
|
| 136 |
+
} else {
|
| 137 |
+
btn.disabled = true;
|
| 138 |
+
status.textContent = 'Voice not supported';
|
| 139 |
+
}
|
| 140 |
+
</script>
|
| 141 |
+
""", unsafe_allow_html=True)
|
| 142 |
+
|
| 143 |
+
# --- Single input field for question ---
|
| 144 |
question = st.text_input("Ask a question about your code", key="sem_question")
|
| 145 |
|
| 146 |
if st.button("Run Semantic Search"):
|