Spaces:
Sleeping
Sleeping
import streamlit as st | |
from clarifai_grpc.channel.clarifai_channel import ClarifaiChannel | |
from clarifai_grpc.grpc.api import resources_pb2, service_pb2, service_pb2_grpc | |
from clarifai_grpc.grpc.api.status import status_code_pb2 | |
# Function to make API call and get text completion | |
def get_text_completion(raw_text): | |
PAT = '3ca5bd8b0f2244eb8d0e4b2838fc3cf1' | |
USER_ID = 'openai' | |
APP_ID = 'chat-completion' | |
MODEL_ID = 'GPT-4' | |
MODEL_VERSION_ID = '182136408b4b4002a920fd500839f2c8' | |
channel = ClarifaiChannel.get_grpc_channel() | |
stub = service_pb2_grpc.V2Stub(channel) | |
metadata = (('authorization', 'Key ' + PAT),) | |
userDataObject = resources_pb2.UserAppIDSet(user_id=USER_ID, app_id=APP_ID) | |
post_model_outputs_response = stub.PostModelOutputs( | |
service_pb2.PostModelOutputsRequest( | |
user_app_id=userDataObject, | |
model_id=MODEL_ID, | |
version_id=MODEL_VERSION_ID, | |
inputs=[ | |
resources_pb2.Input( | |
data=resources_pb2.Data( | |
text=resources_pb2.Text( | |
raw=raw_text | |
) | |
) | |
) | |
] | |
), | |
metadata=metadata | |
) | |
if post_model_outputs_response.status.code != status_code_pb2.SUCCESS: | |
raise Exception(f"Post model outputs failed, status: {post_model_outputs_response.status.description}") | |
output = post_model_outputs_response.outputs[0] | |
return output.data.text.raw | |
# Streamlit app | |
def main(): | |
st.title("Pyresearch gpt App") | |
# Get user input | |
raw_text = st.text_area("Enter text:", "i need to know about narcotics pemishment") | |
# Perform text completion when button is clicked | |
if st.button("Complete Text"): | |
completion_result = get_text_completion(raw_text) | |
st.success("Completion:\n{}".format(completion_result)) | |
# Display HTML content | |
html_content = ''' | |
<div align="center"> | |
<a href="https://github.com/pyresearch/pyresearch" style="text-decoration:none;"> | |
<img src="https://user-images.githubusercontent.com/34125851/226594737-c21e2dda-9cc6-42ef-b4e7-a685fea4a21d.png" width="2%" alt="" /></a> | |
<img src="https://user-images.githubusercontent.com/34125851/226595799-160b0da3-c9e0-4562-8544-5f20460f7cc9.png" width="2%" alt="" /> | |
<a href="https://www.linkedin.com/company/pyresearch/" style="text-decoration:none;"> | |
<img src="https://user-images.githubusercontent.com/34125851/226596446-746ffdd0-a47e-4452-84e3-bf11ec2aa26a.png" width="2%" alt="" /></a> | |
<img src="https://user-images.githubusercontent.com/34125851/226595799-160b0da3-c9e0-4562-8544-5f20460f7cc9.png" width="2%" alt="" /> | |
<a href="https://twitter.com/Noorkhokhar10" style="text-decoration:none;"> | |
<img src="https://user-images.githubusercontent.com/34125851/226599162-9b11194e-4998-440a-ba94-c8a5e1cdc676.png" width="2%" alt="" /></a> | |
<img src="https://user-images.githubusercontent.com/34125851/226595799-160b0da3-c9e0-4562-8544-5f20460f7cc9.png" width="2%" alt="" /> | |
<a href="https://www.youtube.com/@Pyresearch" style="text-decoration:none;"> | |
<img src="https://user-images.githubusercontent.com/34125851/226599904-7d5cc5c0-89d2-4d1e-891e-19bee1951744.png" width="2%" alt="" /></a> | |
<img src="https://user-images.githubusercontent.com/34125851/226595799-160b0da3-c9e0-4562-8544-5f20460f7cc9.png" width="2%" alt="" /> | |
<a href="https://www.facebook.com/Pyresearch" style="text-decoration:none;"> | |
<img src="https://user-images.githubusercontent.com/34125851/226600380-a87a9142-e8e0-4ec9-bf2c-dd6e9da2f05a.png" width="2%" alt="" /></a> | |
<img src="https://user-images.githubusercontent.com/34125851/226595799-160b0da3-c9e0-4562-8544-5f20460f7cc9.png" width="2%" alt="" /> | |
<a href="https://www.instagram.com/pyresearch/" style="text-decoration:none;"> | |
<img src="https://user-images.githubusercontent.com/34125851/226601355-ffe0b597-9840-4e10-bbef-43d6c74b5a9e.png" width="2%" alt="" /></a> | |
</div> | |
''' | |
st.markdown(html_content, unsafe_allow_html=True) | |
if __name__ == "__main__": | |
main() | |