Update app.py

app.py

@@ -23,68 +23,63 @@ with st.form("user_inputs"):
     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")
-    download()
 
-if button and uploaded_file is not None and mcq_count and subject and tone:
+    if button and uploader_file is not None and mcq_count and subject and tone:
         with st.spinner("loading..."):
             try:
-                text=read_file(uploaded_file)
-                #Count tokens and the cost of API call
+                text = read_file(uploader_file)
+                # Count tokens and the cost of the API call
                 with get_openai_callback() as cb:
-                    response=generate_evaluate_chain(
+                    response = generate_evaluate_chain(
                         {
-                        "text": text,
-                        "number": mcq_count,
-                        "subject":subject,
-                        "tone": tone,
-                        "response_json": json.dumps(RESPONSE_JSON)
-                            }
+                            "text": text,
+                            "number": mcq_count,
+                            "subject": subject,
+                            "tone": tone,
+                            "response_json": json.dumps(RESPONSE_JSON)
+                        }
                     )
-                #st.write(response)
-
             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}")
                 if isinstance(response, dict):
-                    #Extract the quiz data from the response
-                    quiz=response.get("quiz", None)
+                    # Extract the quiz data from the response
+                    quiz = response.get("quiz", None)
                     if quiz is not None:
-                        table_data=get_table_data(quiz)
+                        table_data = get_table_data(quiz)
                         if table_data is not None:
-                            df=pd.DataFrame(table_data)
-                            df.index=df.index+1
+                            df = pd.DataFrame(table_data)
+                            df.index = df.index + 1
+                            download(df)
                             st.table(df)
-                            #Display the review in atext box as well
+                            # Display the review in a text box as well
                             st.text_area(label="Review", value=response["review"])
+                            # Call the download function
+                            
                         else:
                             st.error("Error in the table data")
+                    else:
+                        st.write(response)
 
-                else:
-                    st.write(response)
-                    
-def download():
-    if button:
-        csv_filename = "generated_mcqs.csv"
-        pdf_filename = "generated_mcqs.pdf"
-    
-        csv = st.download_button(
-            label="Download as CSV",
-            data=df.to_csv(index=False).encode("utf-8"),
-            file_name=csv_filename,
-            key="csv-download",
-        )
-    
-        pdf = st.download_button(
-            label="Download as PDF",
-            data=df.to_html().encode("utf-8"),
-            file_name=pdf_filename,
-            key="pdf-download",
-        )
+def download(df):
+    csv_filename = "generated_mcqs.csv"
+    pdf_filename = "generated_mcqs.pdf"
 
+    csv = st.download_button(
+        label="Download as CSV",
+        data=df.to_csv(index=False).encode("utf-8"),
+        file_name=csv_filename,
+        key="csv-download",
+    )
 
+    pdf = st.download_button(
+        label="Download as PDF",
+        data=df.to_html().encode("utf-8"),
+        file_name=pdf_filename,
+        key="pdf-download",
+    )