Update app.py

app.py

@@ -23,59 +23,68 @@ 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()
 
-    # Move these lines after the form submission
-    if button:
-        csv_filename = "generated_mcqs.csv"
-        pdf_filename = "generated_mcqs.pdf"
+if button and uploaded_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
+                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)
+                            }
+                    )
+                #st.write(response)
 
-        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}")
+            except Exception as e:
+                traceback.print_exception(type(e), e, e.__traceback__)
+                st.error("Error")
 
-            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)
+            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)
+                    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)
+                            #Display the review in atext box as well
+                            st.text_area(label="Review", value=response["review"])
+                        else:
+                            st.error("Error in the table data")
 
-                        st.text_area(label="Review", value=response["review"])
-                    else:
-                        st.error("Error in the table data")
                 else:
                     st.write(response)
-
-if button:
-    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():
+    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",
+        )