Update app.py

app.py

@@ -45,16 +45,18 @@ if upload_code_file is not None:
     code_image = Image.open(upload_code_file)
     st.image(code_image, caption="Uploaded Code Image", use_column_width=True)
 
-# Prompt Template for code interpretation
-input_prompt_code = """
-You are an expert in coding. The image contains a piece of code. Your task is to interpret the code and solve the problem it presents.
-Provide an analysis of the code, its functionality, and any potential improvements or solutions.
-"""
-
-# Button for code interpretation
-submit = st.button("Interpret Code!")
-if submit:
-    code_image_data = prep_image(upload_code_file)
-    response = get_response_code(code_image_data, input_prompt_code)
-    st.subheader("Code AI: ")
-    st.write(response)
+    # Prompt Template for code interpretation
+    input_prompt_code = """
+    You are an expert in coding. The image contains a piece of code. Your task is to interpret the code and solve the problem it presents.
+    Provide a complete solution of the code.
+    """
+
+    # Button for code interpretation
+    submit = st.button("Interpret Code!")
+    if submit:
+        code_image_data = prep_image(upload_code_file)
+        response = get_response_code(code_image_data, input_prompt_code)
+        st.subheader("Code AI: ")
+        st.write(response)
+else:
+    st.info("Please upload an image of the code to proceed.")