change the api url

app/app.py

@@ -66,6 +66,7 @@ def process(title: str, keywords: str, text: str,
             penalty_alpha = 0.6):
     # st.write("Cache miss: process")
     url = 'https://news-api.uncool.ai/api/text_generator/v1'
+    # url = 'http://localhost:8000/api/text_generator/v1'
     headers = {'Authorization': 'Bearer ' + news_api_auth_token}
     data = {
         "title": title,
@@ -83,9 +84,7 @@ def process(title: str, keywords: str, text: str,
     }
     r = requests.post(url, headers=headers, data=data)
     if r.status_code == 200:
-        result = r.json()['generated_text']
-        title_index = result.find("title: ")
-        result = result[:title_index] if title_index > 0 else result
+        result = r.json()
         return result
     else:
         return "Error: " + r.text
@@ -208,7 +207,7 @@ if prompt_group_name in ["Indonesian Newspaper"]:
     if st.button("Run"):
         with st.spinner(text="Getting results..."):
             memory = psutil.virtual_memory()
-            st.subheader("Result")
+            # st.subheader("Result")
             time_start = time.time()
             # text_generator = MODELS[model_type]["text_generator"]
             result = process(title=session_state.title,
@@ -219,9 +218,15 @@ if prompt_group_name in ["Indonesian Newspaper"]:
             time_end = time.time()
             time_diff = time_end - time_start
             # result = result[0]["generated_text"]
-            st.write(result.replace("\n", "  \n"))
-            st.text("Translation")
-            translation = translate(result, "en", "id")
+            title = f"### {session_state.title}"
+            tldr = f"*{result['description'].strip()}*"
+            caption = f"*Photo Caption: {result['caption'].strip()}*" if result['caption'].strip() != "" else ""
+            st.markdown(title)
+            st.markdown(tldr)
+            st.markdown(result["generated_text"].replace("\n", "  \n"))
+            st.markdown(caption.replace("\n", "  \n"))
+            st.markdown("**Translation**")
+            translation = translate(result["generated_text"], "en", "id")
             st.write(translation.replace("\n", "  \n"))
             # st.write(f"*do_sample: {do_sample}, top_k: {top_k}, top_p: {top_p}, seed: {seed}*")
             info = f"""