add: batch code and batch download code

app.py

@@ -1,12 +1,11 @@
-"""
-Created By: ishwor subedi
-Date: 2024-10-03
-"""
 import os
 import streamlit as st
 from datetime import datetime
 from groq import Groq
 from logic import LLMClient, CodeProcessor
+from batch_code_logic_csv import csv_read_batch_code
+import zipfile
+import io
 
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
@@ -14,32 +13,73 @@ st.title("Code Analysis with LLMs")
 
 st.sidebar.title("Input Options")
 
-code_input_method = st.sidebar.radio("How would you like to provide your code?", ("Paste Code", "Upload Code File"))
+code_input_method = st.sidebar.radio("How would you like to provide your code?",
+                                     ("Upload CSV file", "Upload Code File"))
+
+code_dict = {}
+
+if code_input_method == "Upload CSV file":
+    uploaded_file = st.sidebar.file_uploader("Upload your CSV/Excel file", type=["csv", "xlsx"])
+    if uploaded_file is not None:
+        code_dict = csv_read_batch_code(uploaded_file)  # returns a dict with keys and code snippets
 
-code_text = ""
-if code_input_method == "Paste Code":
-    code_text = st.sidebar.text_area("Paste your code here:")
 elif code_input_method == "Upload Code File":
     uploaded_file = st.sidebar.file_uploader("Upload your code file", type=["py", "txt"])
     if uploaded_file is not None:
         code_text = uploaded_file.read().decode("utf-8")
+        code_dict = {"single_code": code_text}  # Wrap in dict for consistency
+
+model_choice = st.sidebar.selectbox("Select LLM Model",
+                                    ["llama-3.2-90b-text-preview", "llama-3.2-90b-text-preview", "llama3-8b-8192"])
+
+if code_dict:
+    unique_key = st.sidebar.selectbox("Select a Key for Analysis", list(code_dict.keys()))
+
+    if st.sidebar.button("Analyze Code") and unique_key:
+        llm_obj = LLMClient(client)
+        processor = CodeProcessor(llm_obj)
+
+        code_text = code_dict[unique_key]
+        markdown_output = processor.process_code(code_text, model_choice)
+
+        with st.expander(f"Analysis for {unique_key}"):
+            st.markdown(markdown_output)
+
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        st.download_button(
+            label=f"Download {unique_key} Result as Markdown",
+            data=markdown_output,
+            file_name=f"code_analysis_{unique_key}_{timestamp}.md",
+            mime="text/markdown"
+        )
 
-model_choice = st.sidebar.selectbox("Select LLM Model", ["llama-3.2-90b-text-preview", "llama-3.2-90b-text-preview", "llama3-8b-8192"])
+    if st.sidebar.button("Batch Predict"):
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        all_markdowns = {}
+        for key, code_text in code_dict.items():
+            markdown_output = processor.process_code(code_text, model_choice)
+            all_markdowns[key] = markdown_output
 
-if st.sidebar.button("Analyze Code") and code_text:
-    llm_obj = LLMClient(client)
-    processor = CodeProcessor(llm_obj)
+            with st.expander(f"Analysis for {key}"):
+                st.markdown(markdown_output)
 
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    markdown_output = processor.process_code(code_text, model_choice)
+            st.download_button(
+                label=f"Download {key} Result as Markdown",
+                data=markdown_output,
+                file_name=f"code_analysis_{key}_{timestamp}.md",
+                mime="text/markdown"
+            )
 
-    st.markdown(markdown_output)
+        zip_buffer = io.BytesIO()
+        with zipfile.ZipFile(zip_buffer, "w") as zip_file:
+            for key, markdown_output in all_markdowns.items():
+                zip_file.writestr(f"code_analysis_{key}_{timestamp}.md", markdown_output)
 
-    st.download_button(
-        label="Download Result as Markdown",
-        data=markdown_output,
-        file_name=f"code_analysis_{timestamp}.md",
-        mime="text/markdown"
-    )
+        st.download_button(
+            label="Download All as Zip",
+            data=zip_buffer.getvalue(),
+            file_name=f"code_analysis_batch_{timestamp}.zip",
+            mime="application/zip"
+        )
 else:
-    st.write("Please paste or upload your code to analyze.")
+    st.write("Please upload your file to analyze.")

batch_code_logic_csv.py

@@ -0,0 +1,28 @@
+"""
+Created By: ishwor subedi
+Date: 2024-10-04
+"""
+import pandas as pd
+
+
+def csv_read_batch_code(dataframe):
+    dataframe.dropna()
+
+    dataframe['Logic ID'].unique()
+
+    unique_value_indices = {}
+
+    for logic_id in dataframe['Logic ID'].unique():
+        row_indices = dataframe[dataframe['Logic ID'] == logic_id].index.tolist()
+        unique_value_indices[logic_id] = [min(row_indices), max(row_indices)]
+
+    print(unique_value_indices)
+
+    code_dict = {}
+
+    for logic_id, (min_row, max_row) in unique_value_indices.items():
+        codes = dataframe.loc[min_row:max_row, 'Code'].tolist()
+
+        code_dict[logic_id] = codes
+
+    return code_dict

logic.py

@@ -1,7 +1,3 @@
-import os
-from groq import Groq
-
-
 class LLMClient:
     def __init__(self, api_client):
         self.client = api_client