Update app.py

app.py

@@ -3,7 +3,7 @@ import json
 import pandas as pd
 import traceback
 import streamlit as st
-from src.mcqgenerator.utilis import read_file,get_table_data
+from src.mcqgenerator.utilis import read_file, get_table_data
 from src.mcqgenerator.logger import logging
 from src.mcqgenerator.mcqgenerator import generate_evaluate_chain
 from langchain_community.callbacks import get_openai_callback
@@ -18,66 +18,63 @@ with open('Response.json', 'r') as file:
 st.title("MCQs Creator Application")
 
 with st.form("user_inputs"):
-    uploader_file=st.file_uploader("Upload a PDF or Txt file")
+    uploader_file = st.file_uploader("Upload a PDF or Txt file")
+    mcq_count = st.text_input("No. of MCQS")
+    subject = st.text_input("Insert subject", max_chars=20)
+    tone = st.text_input("Complexity Level of Questions", max_chars=20, placeholder="Simple")
+    button = st.form_submit_button("Create MCQs")
 
-    mcq_count=st.text_input("No .of MCQS")
+    # Move these lines after the form submission
+    if button:
+        csv_filename = "generated_mcqs.csv"
+        pdf_filename = "generated_mcqs.pdf"
 
-    subject=st.text_input("Insert subject", max_chars=20)
+        try:
+            text = read_file(uploader_file)
+            with get_openai_callback() as cb:
+                response = generate_evaluate_chain({
+                    "text": text,
+                    "number": mcq_count,
+                    "subject": subject,
+                    "tone": tone,
+                    "response_json": json.dumps(RESPONSE_JSON)
+                })
+        except Exception as e:
+            traceback.print_exception(type(e), e, e.__traceback__)
+            st.error("ERROR")
+        else:
+            print(f"Total Tokens: {cb.total_tokens}")
+            print(f"Prompt Tokens: {cb.prompt_tokens}")
+            print(f"Completion Tokens: {cb.completion_tokens}")
+            print(f"Total Cost: {cb.total_cost}")
 
-    tone=st.text_input("Complexity Level of Questions",max_chars=20, placeholder="Simple")
+            if isinstance(response, dict):
+                quiz = response.get("quiz", None)
+                if quiz is not None:
+                    table_data = get_table_data(quiz)
+                    if table_data is not None:
+                        df = pd.DataFrame(table_data)
+                        df.index = df.index + 1
+                        st.table(df)
 
-    button=st.form_submit_button("Create MCQs")
+                        st.text_area(label="Review", value=response["review"])
+                    else:
+                        st.error("Error in the table data")
+                else:
+                    st.write(response)
 
-    if button and uploader_file is not None and mcq_count and subject and tone:
-        with st.spinner("Loading...."):
-            try:
-                text=read_file(uploader_file)
+        csv = st.download_button(
+            label="Download as CSV",
+            data=df.to_csv(index=False).encode("utf-8"),
+            file_name=csv_filename,
+            key="csv-download",
+        )
 
-                with get_openai_callback() as cb:
-                        response=generate_evaluate_chain(
-                            {
-                                "text": text,
-                                "number": mcq_count,
-                                "subject":subject,
-                                "tone": tone,
-                                "response_json": json.dumps(RESPONSE_JSON)
-                            }
-                 )
-                        
-            except Exception as e:
-                  traceback.print_exception(type(e), e,e.__traceback__)
-                  st.error("ERROR")
+        pdf = st.download_button(
+            label="Download as PDF",
+            data=df.to_html().encode("utf-8"),
+            file_name=pdf_filename,
+            key="pdf-download",
+        )
 
-            else:
-                 print(f"Total Tokens:{cb.total_tokens}")
-                 print(f"Prompt Tokens:{cb.prompt_tokens}")
-                 print(f"Completion Tokens:{cb.completion_tokens}")
-                 print(f"Total Cost:{cb.total_cost}")
-                 if isinstance(response,dict):
-                       
-                       quiz=response.get("quiz",None)
-                       if quiz is not None:
-                            table_data=get_table_data(quiz)
-                            if table_data is not None:
-                                 df=pd.DataFrame(table_data)
-                                 df.index=df.index+1
-                                 st.table(df)
 
-                                 st.text_area(label="Review",value=response["review"])
-
-                                  # Download Buttons for CSV and PDF
-                                 csv_filename = "generated_mcqs.csv"
-                                 csv_link = f'<a href="data:file/csv;base64,{df.to_csv(index=False).encode("utf-8").decode()}" download="{csv_filename}">Download as CSV</a>'
-                                 st.markdown(csv_link, unsafe_allow_html=True)
-    
-                                 pdf_filename = "generated_mcqs.pdf"
-                                 pdf_link = f'<a href="data:application/pdf;base64,{df.to_html().encode("utf-8").decode()}" download="{pdf_filename}">Download as PDF</a>'
-                                 st.markdown(pdf_link, unsafe_allow_html=True)
-
-                            else:
-                                 st.error("Error in the table date")
-
-                       else:
-                            st.write(response)
-
-